Update file model.py

model.py

@@ -55,7 +55,7 @@ class Model:
         model = self.model
         model.eval()
         
-        input_ids = tokenizer(prompt, return_tensors='pt').input_ids
+        input_ids = tokenizer.encode(prompt)
         
         for token_n in range(n_tokens_to_gen):
             with torch.no_grad():