From 3146aff22e0136ee4e6cf13ff1fa1553cedbf133 Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Sat, 4 Apr 2026 14:02:49 +0200 Subject: [PATCH] fix: support huggingface-cli login for community cache uploads Fall back to huggingface_hub stored token so users who ran 'huggingface-cli login' can upload without setting HF_TOKEN manually. Co-Authored-By: Claude Sonnet 4.6 --- lora_optimizer.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/lora_optimizer.py b/lora_optimizer.py index f084ec9..99507d7 100644 --- a/lora_optimizer.py +++ b/lora_optimizer.py @@ -9345,7 +9345,14 @@ class LoRAAutoTuner(LoRAOptimizer): and len(content_hashes) == len(active_loras)): _hf_token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN") if not _hf_token: - logging.warning("[AutoTuner Community] HF_TOKEN not set, skipping upload") + try: + from huggingface_hub import get_token as _hf_get_token + _hf_token = _hf_get_token() + except Exception: + pass + if not _hf_token: + logging.warning("[AutoTuner Community] No HF token found. " + "Run 'huggingface-cli login' or set HF_TOKEN to enable uploads.") else: self._community_upload_results( new_lora_entries, new_pair_entries,