Upload bge_custom_impl.py with huggingface_hub

bge_custom_impl.py

@@ -11,8 +11,8 @@ class ConcatCustomPooling(nn.Module):
         super().__init__()
         self.layers = layers
         self.base_name = model_name_or_path
-        self.tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
-        self.model = AutoModel.from_pretrained(model_name_or_path)
+        self.tokenizer = AutoTokenizer.from_pretrained("BAAI/bge-large-en-v1.5")
+        self.model = AutoModel.from_pretrained("BAAI/bge-large-en-v1.5")
         self.model.eval()
         self.max_seq_len = max_seq_len