Upload bge_custom_impl.py with huggingface_hub

bge_custom_impl.py

@@ -7,7 +7,17 @@ import torch.nn as nn
 from transformers import AutoModel, AutoTokenizer
 
 class ConcatCustomPooling(nn.Module):
-    def __init__(self, model_name_or_path="BAAI/bge-large-en-v1.5",layers=None,max_seq_len=512, **kwargs):
+    def __init__(self, model_name_or_path="BAAI/bge-large-en-v1.5",layers=[
+        15,
+        16,
+        17,
+        18,
+        19,
+        20,
+        21,
+        22,
+        23
+    ],max_seq_len=512, **kwargs):
         super().__init__()
         self.layers = layers
         self.base_name = model_name_or_path