Skip to content

Commit

Permalink
Merge pull request instructlab#359 from bbrowning/remove-llmblock-print
Browse files Browse the repository at this point in the history
Move a spurious print to a debug log message
  • Loading branch information
mergify[bot] authored Nov 11, 2024
2 parents 9173237 + 53bf181 commit 2766095
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 @@ -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
Expand Down

0 comments on commit 2766095

Please sign in to comment.