nomic-ai
/

nomic-embed-text-v1

Sentence Similarity

sentence-transformers

Transformers.js

feature-extraction

text-embeddings-inference

Model card Files Files and versions

zpn commited on Feb 23, 2024

Commit

289f532

·

verified ·

1 Parent(s): c9cf92b

Update modeling_hf_nomic_bert.py

Files changed (1) hide show

modeling_hf_nomic_bert.py +2 -2

modeling_hf_nomic_bert.py CHANGED Viewed

@@ -1055,9 +1055,9 @@ class NomicBertModel(NomicBertPreTrainedModel):
     def forward(
         self,
         input_ids,
-        position_ids=None,
-        token_type_ids=None,
         attention_mask=None,
         return_dict=None,
     ):
         if token_type_ids is None:

     def forward(
         self,
         input_ids,
         attention_mask=None,
+        token_type_ids=None,
+        position_ids=None,
         return_dict=None,
     ):
         if token_type_ids is None: