Spaces:
Running
Running
chore: update something
Browse files
lightweight_embeddings/service.py
CHANGED
@@ -186,11 +186,12 @@ class EmbeddingsService:
|
|
186 |
"provider": "CPUExecutionProvider", # or "CUDAExecutionProvider"
|
187 |
"file_name": info.onnx_file,
|
188 |
},
|
|
|
189 |
)
|
190 |
else:
|
191 |
# Fallback: standard HF loading
|
192 |
self.text_models[t_model_type] = SentenceTransformer(
|
193 |
-
info.model_id, device=self.device
|
194 |
)
|
195 |
|
196 |
for i_model_type in ImageModelType:
|
|
|
186 |
"provider": "CPUExecutionProvider", # or "CUDAExecutionProvider"
|
187 |
"file_name": info.onnx_file,
|
188 |
},
|
189 |
+
trust_remote_code=True,
|
190 |
)
|
191 |
else:
|
192 |
# Fallback: standard HF loading
|
193 |
self.text_models[t_model_type] = SentenceTransformer(
|
194 |
+
info.model_id, device=self.device, trust_remote_code=True,
|
195 |
)
|
196 |
|
197 |
for i_model_type in ImageModelType:
|