WatNeru commited on
Commit
8620bbf
·
1 Parent(s): dbb276c

Add token debugging and explicit login for gated model access

Browse files
Files changed (1) hide show
  1. app.py +15 -0
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),