update load_from_state_dict

modeling_hf_nomic_bert.py

@@ -60,10 +60,14 @@ def state_dict_from_pretrained(model_name, safe_serialization=False, device=None
         load_safe = True
     else:  # Try loading from HF hub instead of from local files
         weight_name = WEIGHTS_NAME if not safe_serialization else SAFE_WEIGHTS_NAME
-        resolved_archive_file = cached_file(model_name, weight_name, _raise_exceptions_for_missing_entries=False)
+        resolved_archive_file = cached_file(
+            model_name, weight_name, token=True, _raise_exceptions_for_missing_entries=False
+        )
         if resolved_archive_file is None:
             weight_index = WEIGHTS_INDEX_NAME if not safe_serialization else SAFE_WEIGHTS_INDEX_NAME
-            resolved_archive_file = cached_file(model_name, weight_index, _raise_exceptions_for_missing_entries=False)
+            resolved_archive_file = cached_file(
+                model_name, weight_index, token=True, _raise_exceptions_for_missing_entries=False
+            )
             if resolved_archive_file is not None:
                 is_sharded = True