Add embedded Hugging Face token fallback

This commit is contained in:
CoolBoy
2026-07-10 01:02:25 +08:00
parent 597d3b1d86
commit 5460fbd108
4 changed files with 10 additions and 6 deletions

View File

@@ -4,7 +4,7 @@ import argparse
import os
from pathlib import Path
from defaults import EMBEDDED_MODELHUB_XC_TOKEN
from defaults import EMBEDDED_HF_TOKEN, EMBEDDED_MODELHUB_XC_TOKEN
DEFAULT_KEY_PATH = Path("KEY.md")
@@ -43,6 +43,8 @@ def ensure_tokens(args: argparse.Namespace) -> None:
if not getattr(args, "hf_token", None):
args.hf_token = os.getenv("HF_TOKEN") or values.get("HF_TOKEN")
if not getattr(args, "hf_token", None):
args.hf_token = EMBEDDED_HF_TOKEN
modelhub_token = getattr(args, "modelhub_token", None)
if not modelhub_token: