Fix: download mapping for scripts/train.py

runners/run_5090.py

@@ -45,12 +45,18 @@ os.makedirs("checkpoints", exist_ok=True)
 
 # 3. Fetch Architecture & Engine from ViuAI/ViuTranslate
 print("\n📥 [2/5] Fetching model architecture and training engine from ViuAI/ViuTranslate...")
-code_files = ["model.py", "config.py", "tokenizer.json", "train_viutranslate.py", "inference.py"]
-for f in code_files:
-    if not os.path.exists(f):
-        print(f"   • Fetching {f}...")
-        p = hf_hub_download(repo_id="ViuAI/ViuTranslate", filename=f, token=HF_TOKEN)
-        shutil.copy(p, f)
+code_files = [
+    ("model.py", "model.py"),
+    ("config.py", "config.py"),
+    ("tokenizer.json", "tokenizer.json"),
+    ("scripts/train.py", "train.py"),
+    ("scripts/inference.py", "inference.py"),
+]
+for src, dst in code_files:
+    if not os.path.exists(dst):
+        print(f"   • Fetching {src} -> {dst}...")
+        p = hf_hub_download(repo_id="ViuAI/ViuTranslate", filename=src, token=HF_TOKEN)
+        shutil.copy(p, dst)
 print("   ✅ Architecture files ready.")
 
 # 4. Fetch 100K+ Real Dataset Shards from ViuAI/ViuTranslate-Data
@@ -81,7 +87,7 @@ print("   ✅ Base weights ready.")
 # bf16 enabled, Fused AdamW, TF32 matmuls
 print("\n⚡ [5/5] Launching RTX 5090 Ultra-Tuned Training...")
 cmd = [
-    sys.executable, "train_viutranslate.py",
+    sys.executable, "train.py",
     "--data_dir", "data/tokenized",
     "--base_ckpt", base_ckpt,
     "--output_dir", "checkpoints",