Update optimum_encoder.py

optimum_encoder.py

@@ -100,7 +100,7 @@ class OptimumEncoder(BaseEncoder):
         for i in range(3):
             output = ort_model(**encoded_input)
 
-        return tokenizer, model
+        return tokenizer, ort_model
 
     def __call__(
         self,