Update app.py
Browse files
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=
|
| 73 |
-
hf_hub_download(repo_id=MODEL_REPO_ID, filename=TEACHER_MODEL, repo_type="model", local_dir=".", 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):
|