Update optimum_encoder.py
Browse files- optimum_encoder.py +1 -1
optimum_encoder.py
CHANGED
@@ -26,7 +26,7 @@ class OptimumEncoder(BaseEncoder):
|
|
26 |
|
27 |
def _initialize_hf_model(self):
|
28 |
try:
|
29 |
-
|
30 |
from optimum.onnxruntime import ORTModelForFeatureExtraction
|
31 |
except ImportError:
|
32 |
raise ImportError(
|
|
|
26 |
|
27 |
def _initialize_hf_model(self):
|
28 |
try:
|
29 |
+
import onnxruntime as ort
|
30 |
from optimum.onnxruntime import ORTModelForFeatureExtraction
|
31 |
except ImportError:
|
32 |
raise ImportError(
|