mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-26 02:33:23 +08:00
Remove unnecessary return
This commit is contained in:
parent
4549dcbbd0
commit
d7d7aabded
@ -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:
|
||||||
|
Loading…
Reference in New Issue
Block a user