Spaces:

GameScribes
/

Multipurpose-AI-Agent-Development

Sleeping

devve1 commited on Jul 9

Commit

846c662

•

1 Parent(s): 7c4c108

Update optimum_encoder.py

Files changed (1) hide show

optimum_encoder.py CHANGED Viewed

@@ -70,7 +70,7 @@ class OptimumEncoder(BaseEncoder):
         ort_model = ORTModelForFeatureExtraction.from_pretrained(
             model_id=self.name,
-            model_type='model_fp16',
             subfolder='onnx',
             provider=['TensorrtExecutionProvider'],
             provider_options=provider_options,

         ort_model = ORTModelForFeatureExtraction.from_pretrained(
             model_id=self.name,
+            file_name='model_fp16.onnx',
             subfolder='onnx',
             provider=['TensorrtExecutionProvider'],
             provider_options=provider_options,