Update optimum_encoder.py

optimum_encoder.py

@@ -98,7 +98,7 @@ class OptimumEncoder(BaseEncoder):
         ).to(self.device)
         
         for i in range(3):
-            output = ort_model.generate(**encoded_input)
+            output = ort_model(**encoded_input)
 
         return tokenizer, model