lamhieu commited on
Commit
f9ce04f
·
1 Parent(s): de24ee4

chore: update something

Browse files
Files changed (1) hide show
  1. lightweight_embeddings/service.py +2 -1
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: