pinecone
/

ConstBERT

Feature Extraction

text-embeddings-inference

Model card Files Files and versions Community

caesar-one commited on May 14

Commit

cb5f0b7

·

verified ·

1 Parent(s): b5eb6ef

Upload ConstBERT

Files changed (1) hide show

modeling.py +2 -2

modeling.py CHANGED Viewed

@@ -90,10 +90,10 @@ class ConstBERT(BertPreTrainedModel):
     @classmethod
-    def from_pretrained(cls, name_or_path):
         colbert_config = ColBERTConfig(name_or_path)
         colbert_config = ColBERTConfig.from_existing(ColBERTConfig.load_from_checkpoint(name_or_path), colbert_config)
-        obj = super().from_pretrained(name_or_path, colbert_config=colbert_config)
         obj.base = name_or_path
         return obj

     @classmethod
+    def from_pretrained(cls, name_or_path, config=None):
         colbert_config = ColBERTConfig(name_or_path)
         colbert_config = ColBERTConfig.from_existing(ColBERTConfig.load_from_checkpoint(name_or_path), colbert_config)
+        obj = super().from_pretrained(name_or_path, colbert_config=colbert_config, config=config)
         obj.base = name_or_path
         return obj