Skip to content

Commit

Permalink
Move a spurious print to a debug log message
Browse files Browse the repository at this point in the history
Signed-off-by: Ben Browning <[email protected]>
  • Loading branch information
bbrowning committed Nov 10, 2024
1 parent e0698d6 commit 53bf181
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/instructlab/sdg/llmblock.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 53bf181

Please sign in to comment.