pinecone
/

ConstBERT

Feature Extraction

text-embeddings-inference

Model card Files Files and versions

caesar-one commited on May 14

Commit

96f6622

·

verified ·

1 Parent(s): cb5f0b7

Upload ConstBERT

Files changed (1) hide show

modeling.py +1 -1

modeling.py CHANGED Viewed

@@ -90,7 +90,7 @@ class ConstBERT(BertPreTrainedModel):
     @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)

     @classmethod
+    def from_pretrained(cls, name_or_path, config=None, *args, **kwargs):
         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)