nomic-ai
/

nomic-bert-2048

Model card Files Files and versions Community

zpn commited on May 2

Commit

b476a30

•

1 Parent(s): 2c2561b

Update modeling_hf_nomic_bert.py

Files changed (1) hide show

modeling_hf_nomic_bert.py +1 -1

modeling_hf_nomic_bert.py CHANGED Viewed

@@ -353,7 +353,7 @@ class NomicBertPreTrainedModel(PreTrainedModel):
             load_return = model.load_state_dict(state_dict, strict=False)
         else:
             # TODO: can probably check config class and see if we need to remap from a bert model
-            state_dict = state_dict_from_pretrained(model_name, safe_serialization=kwargs.get("safe_serialization", False))
             state_dict = remap_bert_state_dict(
                 state_dict,
                 config,

             load_return = model.load_state_dict(state_dict, strict=False)
         else:
             # TODO: can probably check config class and see if we need to remap from a bert model
+            state_dict = state_dict_from_pretrained(model_name)
             state_dict = remap_bert_state_dict(
                 state_dict,
                 config,