Fix: force_download=True to bypass HF cache, add assertion checks

IRIS_Training_Notebook.ipynb

@@ -81,11 +81,18 @@
     "from huggingface_hub import hf_hub_download\n",
     "import shutil\n",
     "\n",
+    "# Force fresh download (bypass cache) to get latest version\n",
     "for f in [\"iris_model.py\"]:\n",
-    "    shutil.copy(hf_hub_download(\"asdf98/IRIS-architecture\", f), f\"./{f}\")\n",
+    "    path = hf_hub_download(\"asdf98/IRIS-architecture\", f, force_download=True)\n",
+    "    shutil.copy(path, f\"./{f}\")\n",
     "\n",
     "from iris_model import IRIS, IRISConfig, IRISGenerator, create_iris_tiny, create_iris_small, count_parameters\n",
-    "print(\"\u2705 IRIS loaded\")"
+    "print(\"\u2705 IRIS loaded\")\n",
+    "\n",
+    "# Quick verification that train_step_latent exists\n",
+    "assert hasattr(IRIS, 'train_step_latent'), \"ERROR: Old iris_model.py cached! Restart runtime and re-run.\"\n",
+    "assert hasattr(IRIS, 'generate_latent'), \"ERROR: Old iris_model.py cached! Restart runtime and re-run.\"\n",
+    "print(\"\u2705 Verified: train_step_latent and generate_latent available\")"
    ],
    "outputs": [],
    "execution_count": null