diff --git a/phi2/phi2.py b/phi2/phi2.py index ede79ea2..2d3f792a 100644 --- a/phi2/phi2.py +++ b/phi2/phi2.py @@ -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: