Remove unnecessary return

This commit is contained in:
devonthomas35 2023-12-14 15:52:22 -08:00 committed by GitHub
parent 4549dcbbd0
commit d7d7aabded
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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: