diff --git a/src/instructlab/sdg/llmblock.py b/src/instructlab/sdg/llmblock.py index a081c597..0e9a5f22 100644 --- a/src/instructlab/sdg/llmblock.py +++ b/src/instructlab/sdg/llmblock.py @@ -191,7 +191,7 @@ def _gen_kwargs(self, max_num_token_override, gen_kwargs, **defaults): def _generate(self, samples) -> list: prompts = [self._format_prompt(sample) for sample in samples] logger.debug(f"STARTING GENERATION FOR LLMBlock USING PROMPTS: {prompts}") - print(self.gen_kwargs) + logger.debug(f"Generation arguments: {self.gen_kwargs}") if self.server_supports_batched: response = self.ctx.client.completions.create( prompt=prompts, **self.gen_kwargs