From 8cd8b0ec1409cdd90da8e4e659a3dc7ee89f6826 Mon Sep 17 00:00:00 2001 From: Lukas Kuhn Date: Tue, 3 Sep 2024 10:19:15 +0200 Subject: [PATCH] Fixed docs so the model loads correctly --- docs/example_hres_t0.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/example_hres_t0.ipynb b/docs/example_hres_t0.ipynb index 1ee291b..7c98d0d 100644 --- 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",