hoangth2k4
/

emb_onnx

Model card Files Files and versions

hoangth2k4 commited on Feb 6

Commit

d39b79f

·

verified ·

1 Parent(s): 6ccc493

Update bgem3_model.py

Files changed (1) hide show

bgem3_model.py +3 -1

bgem3_model.py CHANGED Viewed

@@ -27,7 +27,9 @@ class BGEM3InferenceModel(nn.Module):
         )
         self.config = AutoConfig.from_pretrained(model_name)
-        self.model = AutoModel.from_pretrained(model_name)
         self.colbert_linear = torch.nn.Linear(
             in_features=self.model.config.hidden_size,
             out_features=(

         )
         self.config = AutoConfig.from_pretrained(model_name)
+        self.model = AutoModel.from_pretrained(model_name,
+                                                trust_remote_code=True,
+                                                use_safetensors=True)
         self.colbert_linear = torch.nn.Linear(
             in_features=self.model.config.hidden_size,
             out_features=(