Upload train_tiny.py with huggingface_hub

train_tiny.py

@@ -228,6 +228,14 @@ def train():
                 if "token_embed" in name or "lm_head" in name:
                     param.requires_grad = True
 
+    # ── torch.compile ──────────────────────────────────────────────────────
+    if t_cfg.get("compile", False):
+        try:
+            model = torch.compile(model, dynamic=True)
+            print("[Compile] torch.compile enabled (dynamic=True)")
+        except Exception as e:
+            print(f"[Compile] Failed: {e} — continuing without compile")
+
     hf_repo = d_cfg.get("hf_repo")
     is_streaming = bool(hf_repo)