From 205ba0c97bf5f11c486705ec5cbbc039ac7ab511 Mon Sep 17 00:00:00 2001
From: Anushan Fernando <35841118+Nush395@users.noreply.github.com>
Date: Fri, 5 Apr 2024 14:40:33 +0100
Subject: [PATCH] Update sampling_tutorial.ipynb

Fix a small typo in sampling tutorial.
---
 colabs/sampling_tutorial.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colabs/sampling_tutorial.ipynb b/colabs/sampling_tutorial.ipynb
index 9563adf..3eb0f46 100644
--- a/colabs/sampling_tutorial.ipynb
+++ b/colabs/sampling_tutorial.ipynb
@@ -107,7 +107,7 @@
     "VARIANT = '2b-it' # @param ['2b', '2b-it', '7b', '7b-it'] {type:\"string\"}\n",
     "weights_dir = kagglehub.model_download(f'google/gemma/Flax/{VARIANT}')\n",
     "\n",
-    "ckpt_path = os.path.join(weights_dir, variant)\n",
+    "ckpt_path = os.path.join(weights_dir, VARIANT)\n",
     "vocab_path = os.path.join(weights_dir, 'tokenizer.model')"
    ]
   },