[Misc] format and refactor some examples (#16252)

Signed-off-by: reidliu41 <reid201711@gmail.com>
Co-authored-by: reidliu41 <reid201711@gmail.com>
This commit is contained in:
Reid
2025-04-08 18:42:32 +08:00
committed by GitHub
parent 995e3d1f41
commit 7f00899ff7
13 changed files with 178 additions and 115 deletions

View File

@@ -1089,14 +1089,18 @@ def main(args):
start_time = time.time()
outputs = llm.generate(inputs, sampling_params=sampling_params)
elapsed_time = time.time() - start_time
print("-" * 50)
print("-- generate time = {}".format(elapsed_time))
print("-" * 50)
else:
outputs = llm.generate(inputs, sampling_params=sampling_params)
print("-" * 50)
for o in outputs:
generated_text = o.outputs[0].text
print(generated_text)
print("-" * 50)
if __name__ == "__main__":