Fill-Mask
Transformers
PyTorch
Safetensors
English
nomic_bert
custom_code
zpn jxm commited on
Commit
53c3a30
1 Parent(s): a971780

make input_ids optional (#8)

Browse files

- make input_ids optional (90fe1be81ee87cbbcc7a39b1d26e7252b5ff69b6)


Co-authored-by: Jack Morris <jxm@users.noreply.huggingface.co>

Files changed (1) hide show
  1. modeling_hf_nomic_bert.py +1 -1
modeling_hf_nomic_bert.py CHANGED
@@ -1678,7 +1678,7 @@ class NomicBertModel(NomicBertPreTrainedModel):
1678
 
1679
  def forward(
1680
  self,
1681
- input_ids,
1682
  attention_mask=None,
1683
  position_ids=None,
1684
  token_type_ids=None,
 
1678
 
1679
  def forward(
1680
  self,
1681
+ input_ids=None,
1682
  attention_mask=None,
1683
  position_ids=None,
1684
  token_type_ids=None,