From 53bf1819865c913cf65a508d4c009b993fcafdac Mon Sep 17 00:00:00 2001 From: Ben Browning Date: Sat, 9 Nov 2024 21:26:48 -0500 Subject: [PATCH] Move a spurious print to a debug log message Signed-off-by: Ben Browning --- src/instructlab/sdg/llmblock.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/instructlab/sdg/llmblock.py b/src/instructlab/sdg/llmblock.py index ddb85477..a89c8252 100644 --- a/src/instructlab/sdg/llmblock.py +++ b/src/instructlab/sdg/llmblock.py @@ -175,7 +175,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