From 402c775796652f41320e315a459e3ad7ccad6c64 Mon Sep 17 00:00:00 2001
From: Dibya Ghosh <dibya.ghosh@berkeley.edu>
Date: Wed, 13 Dec 2023 09:22:35 -0600
Subject: [PATCH] Merge pull request #1 from rail-berkeley/fix_colab

Fixes several issues in the inference colab
---
 examples/01_inference_pretrained.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/01_inference_pretrained.ipynb b/examples/01_inference_pretrained.ipynb
index bc44a86f..8dd6c8d5 100644
--- 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\")"
    ]
   },
   {