Fixed docs so the model loads correctly (#22)

microsoft · Sep 3, 2024 · c818005 · c818005
1 parent 08ac374
commit c818005
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/example_hres_t0.ipynb b/docs/example_hres_t0.ipynb
@@ -248,7 +248,7 @@
     "from aurora import Aurora, rollout\n",
     "\n",
     "model = Aurora()\n",
-    "model.load_checkpoint(\"wbruinsma/aurora\", \"aurora-0.25-finetuned.ckpt\")\n",
+    "model.load_checkpoint(\"microsoft/aurora\", \"aurora-0.25-finetuned.ckpt\")\n",
     "\n",
     "model.eval()\n",
     "model = model.to(\"cuda\")\n",