Remove unnecessary return

This commit is contained in:
devonthomas35 2023-12-14 15:52:22 -08:00 committed by GitHub
parent 8d496ba61a
commit 7f992db5bc

View File

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