deploy: 04a5ca0

microsoft · Oct 7, 2024 · 2568fc0 · 2568fc0
1 parent a54896a
commit 2568fc0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/_sources/example_hres_0.1.ipynb b/_sources/example_hres_0.1.ipynb
@@ -210,7 +210,7 @@
     "    # Use manually downloaded and regridded static variables.\n",
     "    surf_stats={\"z\": (-3.270407e03, 6.540335e04)},\n",
     ")\n",
-    "model.load_checkpoint(\"wbruinsma/aurora\", \"aurora-0.1-finetuned.ckpt\")\n",
+    "model.load_checkpoint(\"microsoft/aurora\", \"aurora-0.1-finetuned.ckpt\")\n",
     "\n",
     "model.eval()\n",
     "model = model.to(\"cuda\")\n",

diff --git a/example_hres_0.1.html b/example_hres_0.1.html
@@ -569,7 +569,7 @@ <h2>Loading and Running the Model<a class="headerlink" href="#loading-and-runnin
     <span class="c1"># Use manually downloaded and regridded static variables.</span>
     <span class="n">surf_stats</span><span class="o">=</span><span class="p">{</span><span class="s2">&quot;z&quot;</span><span class="p">:</span> <span class="p">(</span><span class="o">-</span><span class="mf">3.270407e03</span><span class="p">,</span> <span class="mf">6.540335e04</span><span class="p">)},</span>
 <span class="p">)</span>
-<span class="n">model</span><span class="o">.</span><span class="n">load_checkpoint</span><span class="p">(</span><span class="s2">&quot;wbruinsma/aurora&quot;</span><span class="p">,</span> <span class="s2">&quot;aurora-0.1-finetuned.ckpt&quot;</span><span class="p">)</span>
+<span class="n">model</span><span class="o">.</span><span class="n">load_checkpoint</span><span class="p">(</span><span class="s2">&quot;microsoft/aurora&quot;</span><span class="p">,</span> <span class="s2">&quot;aurora-0.1-finetuned.ckpt&quot;</span><span class="p">)</span>
 
 <span class="n">model</span><span class="o">.</span><span class="n">eval</span><span class="p">()</span>
 <span class="n">model</span> <span class="o">=</span> <span class="n">model</span><span class="o">.</span><span class="n">to</span><span class="p">(</span><span class="s2">&quot;cuda&quot;</span><span class="p">)</span>