nomic-ai
/

nomic-bert-2048

Model card Files Files and versions Community

make input_ids optional

#8

by jxm - opened Oct 1

base: refs/heads/main

←

from: refs/pr/8

Discussion Files changed

Files changed (1) hide show

modeling_hf_nomic_bert.py +1 -1

modeling_hf_nomic_bert.py CHANGED Viewed

@@ -1678,7 +1678,7 @@ class NomicBertModel(NomicBertPreTrainedModel):
     def forward(
         self,
-        input_ids,
         attention_mask=None,
         position_ids=None,
         token_type_ids=None,

     def forward(
         self,
+        input_ids=None,
         attention_mask=None,
         position_ids=None,
         token_type_ids=None,