Update modeling_hf_nomic_bert.py
Browse files
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,
|
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]:
|