Fill-Mask
Transformers
PyTorch
Safetensors
English
nomic_bert
custom_code
zpn commited on
Commit
7b260c5
1 Parent(s): 1fe3d3f

Update modeling_hf_nomic_bert.py

Browse files
Files changed (1) hide show
  1. modeling_hf_nomic_bert.py +1 -1
modeling_hf_nomic_bert.py CHANGED
@@ -62,7 +62,7 @@ def state_dict_from_pretrained(model_name, safe_serialization=False, device=None
62
  resolved_archive_file = None
63
  for weight_name in [WEIGHTS_NAME, SAFE_WEIGHTS_NAME, WEIGHTS_INDEX_NAME, SAFE_WEIGHTS_INDEX_NAME]:
64
  resolved_archive_file = cached_file(
65
- model_name, weight_name, token=True, _raise_exceptions_for_missing_entries=False
66
  )
67
  if resolved_archive_file is not None:
68
  if weight_name in [SAFE_WEIGHTS_NAME, SAFE_WEIGHTS_INDEX_NAME]:
 
62
  resolved_archive_file = None
63
  for weight_name in [WEIGHTS_NAME, SAFE_WEIGHTS_NAME, WEIGHTS_INDEX_NAME, SAFE_WEIGHTS_INDEX_NAME]:
64
  resolved_archive_file = cached_file(
65
+ model_name, weight_name, _raise_exceptions_for_missing_entries=False
66
  )
67
  if resolved_archive_file is not None:
68
  if weight_name in [SAFE_WEIGHTS_NAME, SAFE_WEIGHTS_INDEX_NAME]: