Merge pull request #1 from rail-berkeley/fix_colab

Fixes several issues in the inference colab
octo-models · Dec 22, 2023 · 7be5f6a · 7be5f6a
1 parent f0cd903
commit 7be5f6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/01_inference_pretrained.ipynb b/examples/01_inference_pretrained.ipynb
@@ -21,7 +21,7 @@
    "source": [
     "from octo.model.octo_model import OctoModel\n",
     "\n",
-    "model = OctoModel.load_pretrained(\"hf://rail-berkeley/octo-small\")"
+    "model = OCTOModel.load_pretrained(\"hf://rail-berkeley/octo-small\")"
    ]
   },
   {