Add token debugging and explicit login for gated model access
Browse files
app.py
CHANGED
|
@@ -73,6 +73,10 @@ def ensure_model_available() -> str:
|
|
| 73 |
print(f"[MODEL] ensure_model_available() 開始")
|
| 74 |
print(f"[MODEL] モデルリポジトリ: {HF_MODEL_REPO}")
|
| 75 |
print(f"[MODEL] HF_TOKEN設定: {'あり' if HF_TOKEN else 'なし'}")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 76 |
|
| 77 |
# モデルディレクトリのパスを構築(リポジトリ名から)
|
| 78 |
model_dir_name = HF_MODEL_REPO.split("/")[-1] # "Llama-3.2-3B-Instruct"
|
|
@@ -97,7 +101,18 @@ def ensure_model_available() -> str:
|
|
| 97 |
try:
|
| 98 |
if not HF_TOKEN:
|
| 99 |
print("[MODEL] 警告: HF_TOKEN が設定されていません。認証が必要なモデルの場合、ダウンロードに失敗する可能性があります。")
|
|
|
|
| 100 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 101 |
downloaded_dir = snapshot_download(
|
| 102 |
repo_id=HF_MODEL_REPO,
|
| 103 |
local_dir=str(HF_LOCAL_DIR),
|
|
|
|
| 73 |
print(f"[MODEL] ensure_model_available() 開始")
|
| 74 |
print(f"[MODEL] モデルリポジトリ: {HF_MODEL_REPO}")
|
| 75 |
print(f"[MODEL] HF_TOKEN設定: {'あり' if HF_TOKEN else 'なし'}")
|
| 76 |
+
if HF_TOKEN:
|
| 77 |
+
# トークンの最初の数文字だけを表示(セキュリティのため)
|
| 78 |
+
token_preview = HF_TOKEN[:7] + "..." + HF_TOKEN[-4:] if len(HF_TOKEN) > 11 else "***"
|
| 79 |
+
print(f"[MODEL] HF_TOKENプレビュー: {token_preview} (長さ: {len(HF_TOKEN)})")
|
| 80 |
|
| 81 |
# モデルディレクトリのパスを構築(リポジトリ名から)
|
| 82 |
model_dir_name = HF_MODEL_REPO.split("/")[-1] # "Llama-3.2-3B-Instruct"
|
|
|
|
| 101 |
try:
|
| 102 |
if not HF_TOKEN:
|
| 103 |
print("[MODEL] 警告: HF_TOKEN が設定されていません。認証が必要なモデルの場合、ダウンロードに失敗する可能性があります。")
|
| 104 |
+
raise ValueError("HF_TOKEN が設定されていません")
|
| 105 |
|
| 106 |
+
# huggingface_hub の login を使って明示的に認証(念のため)
|
| 107 |
+
try:
|
| 108 |
+
from huggingface_hub import login
|
| 109 |
+
print("[MODEL] huggingface_hub.login() を実行中...")
|
| 110 |
+
login(token=HF_TOKEN, add_to_git_credential=False)
|
| 111 |
+
print("[MODEL] ログイン成功")
|
| 112 |
+
except Exception as login_error:
|
| 113 |
+
print(f"[MODEL] ログインエラー(続行): {login_error}")
|
| 114 |
+
|
| 115 |
+
print(f"[MODEL] snapshot_download を開始: {HF_MODEL_REPO}")
|
| 116 |
downloaded_dir = snapshot_download(
|
| 117 |
repo_id=HF_MODEL_REPO,
|
| 118 |
local_dir=str(HF_LOCAL_DIR),
|