ffzeroHua commited on
Commit
64150db
·
verified ·
1 Parent(s): 03b7169

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -34,6 +34,7 @@ except ImportError:
34
  DATASET_REPO = "AstraNASA/tenhou-scc" # 存放打包好的 DAgger 数据的仓库
35
  MODEL_REPO_ID = "ffzeroHua/Riichi-Model-Repo"
36
  HF_TOKEN = os.getenv("HF_TOKEN")
 
37
  WORKER_ID = os.getenv("WORKER_ID", str(uuid.uuid4())[:6])
38
 
39
  # 🚀 对战双方权重
@@ -69,8 +70,8 @@ EVAL_RUNNING = True
69
  def sync_models_from_hub():
70
  if HF_TOKEN and "你的用户名" not in MODEL_REPO_ID:
71
  print(f"☁️ 正在从 Hub 拉取模型...")
72
- hf_hub_download(repo_id=MODEL_REPO_ID, filename=STUDENT_MODEL, repo_type="model", local_dir=".", token=HF_TOKEN)
73
- hf_hub_download(repo_id=MODEL_REPO_ID, filename=TEACHER_MODEL, repo_type="model", local_dir=".", token=HF_TOKEN)
74
  print("✅ 模型就绪!")
75
 
76
  def patch_event_fast(event_str):
 
34
  DATASET_REPO = "AstraNASA/tenhou-scc" # 存放打包好的 DAgger 数据的仓库
35
  MODEL_REPO_ID = "ffzeroHua/Riichi-Model-Repo"
36
  HF_TOKEN = os.getenv("HF_TOKEN")
37
+ HF_TOKEN_2 = os.getenv("HF_TOKEN_2")
38
  WORKER_ID = os.getenv("WORKER_ID", str(uuid.uuid4())[:6])
39
 
40
  # 🚀 对战双方权重
 
70
  def sync_models_from_hub():
71
  if HF_TOKEN and "你的用户名" not in MODEL_REPO_ID:
72
  print(f"☁️ 正在从 Hub 拉取模型...")
73
+ hf_hub_download(repo_id=MODEL_REPO_ID, filename=STUDENT_MODEL, repo_type="model", local_dir=".", token=HF_TOKEN_2)
74
+ hf_hub_download(repo_id=MODEL_REPO_ID, filename=TEACHER_MODEL, repo_type="model", local_dir=".", token=HF_TOKEN_2)
75
  print("✅ 模型就绪!")
76
 
77
  def patch_event_fast(event_str):