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:
|