From d7d7aabded3b52d36c5f3a3675553d5651639b6f Mon Sep 17 00:00:00 2001
From: devonthomas35 <30363743+devonthomas35@users.noreply.github.com>
Date: Thu, 14 Dec 2023 15:52:22 -0800
Subject: [PATCH] Remove unnecessary return

---
 phi2/phi2.py | 1 -
 1 file changed, 1 deletion(-)

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: