Replace token retrieval

huggingface · Apr 12, 2022 · 33666dd · 33666dd
1 parent f1e86c4
commit 33666dd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/huggingface_hub/hf_api.py b/src/huggingface_hub/hf_api.py
@@ -1178,8 +1178,8 @@ def model_info(
         Returns:
             [`huggingface_hub.hf_api.ModelInfo`]: The model repository information.
         """
-        if token is None:
-            token = HfFolder.get_token()
+
+        token, _ = self._validate_or_retrieve_token(token, function_name="model_info")
 
         path = (
             f"{self.endpoint}/api/models/{repo_id}"