use huggingface path for model weights

Clay-foundation · Mar 22, 2024 · 05a8b2e · 05a8b2e
1 parent ffe1421
commit 05a8b2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/patch_level_cloud_cover.ipynb b/docs/patch_level_cloud_cover.ipynb
@@ -380,7 +380,7 @@
    "outputs": [],
    "source": [
     "DATA_DIR = \"data/minicubes_cloud\"\n",
-    "CKPT_PATH = \"/home/ubuntu/data/checkpoints/mae_epoch-24_val-loss-0.46.ckpt\"\n",
+    "CKPT_PATH = \"https://huggingface.co/made-with-clay/Clay/resolve/main/Clay_v0.1_epoch-24_val-loss-0.46.ckpt\"\n",
     "\n",
     "# Load model\n",
     "multi_model = CLAYModule.load_from_checkpoint(\n",