Ensure backward compatibility for HUGGING_FACE_HUB_TOKEN env variable (…

…#1795)
huggingface · Nov 2, 2023 · 2dc5620 · 2dc5620
1 parent f590bdf
commit 2dc5620
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/huggingface_hub/utils/_hf_folder.py b/src/huggingface_hub/utils/_hf_folder.py
@@ -64,6 +64,8 @@ def get_token(cls) -> Optional[str]:
 
         # 1. Is it set by environment variable ?
         token: Optional[str] = os.environ.get("HF_TOKEN")
+        if token is None:  # Ensure backward compatibility but doesn't have priority
+            token = os.environ.get("HUGGING_FACE_HUB_TOKEN")
         if token is not None:
             token = token.replace("\r", "").replace("\n", "").strip()
             return token