Remove unnecessary return

This commit is contained in:
devonthomas35
2023-12-14 15:52:22 -08:00
committed by GitHub
parent 4549dcbbd0
commit d7d7aabded

View File

@@ -202,7 +202,6 @@ if __name__ == "__main__":
tokens = []
for token, _ in zip(generate(prompt, model), range(args.max_tokens)):
if token == tokenizer.eos_token_id:
break
else: