nomic-ai
/

nomic-embed-text-v1-ablated

Sentence Similarity

sentence-transformers

feature-extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

zpn commited on Jan 15

Commit

dcc6549

•

1 Parent(s): fe45803

Upload model

Files changed (1) hide show

modeling_hf_nomic_bert.py +1 -1

modeling_hf_nomic_bert.py CHANGED Viewed

@@ -330,7 +330,7 @@ class NomicBertPreTrainedModel(PreTrainedModel):
         if "add_pooling_layer" in kwargs:
             model = cls(config, *inputs, add_pooling_layer=kwargs.pop("add_pooling_layer"))
         else:
-            model = cls(config, *inputs)
         # TODO: fix this
         # Assuming we know what we're doing when loading from disk
         # Prob a bad assumption but i'm tired and want to train this asap

         if "add_pooling_layer" in kwargs:
             model = cls(config, *inputs, add_pooling_layer=kwargs.pop("add_pooling_layer"))
         else:
+            model = cls(config, *inputs, add_pooling_layer=False)
         # TODO: fix this
         # Assuming we know what we're doing when loading from disk
         # Prob a bad assumption but i'm tired and want to train this asap