This commit is contained in:
GeeeekExplorer
2025-06-11 21:12:57 +08:00
parent b98e1ca305
commit 386290d69e
8 changed files with 31 additions and 35 deletions

View File

@@ -21,9 +21,9 @@ prompts = [
)
for prompt in prompts
]
completions = llm.generate(prompts, sampling_params)
outputs = llm.generate(prompts, sampling_params)
for p, c in zip(prompts, completions):
for prompt, output in zip(prompts, outputs):
print("\n")
print(f"Prompt: {p}")
print(f"Completion: {c}")
print(f"Prompt: {prompt}")
print(f"Completion: {output["text"]}")