From fbc586b62b94e6bac42229cc617331cd98a50cba Mon Sep 17 00:00:00 2001
From: KumarGitesh2024 <161804283+KumarGitesh2024@users.noreply.github.com>
Date: Wed, 28 Aug 2024 13:48:44 +0530
Subject: [PATCH] Update fine_tuning_tutorial.ipynb

Fixing the NameError issue in the fine _tuning_tutorial
---
 colabs/fine_tuning_tutorial.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colabs/fine_tuning_tutorial.ipynb b/colabs/fine_tuning_tutorial.ipynb
index 9bfc9b4..c312b46 100644
--- a/colabs/fine_tuning_tutorial.ipynb
+++ b/colabs/fine_tuning_tutorial.ipynb
@@ -112,7 +112,7 @@
         "\n",
         "VARIANT = '2b-it' # @param ['2b', '2b-it'] {type:\"string\"}\n",
         "weights_dir = kagglehub.model_download(f'google/gemma/Flax/{VARIANT}')\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')"
       ]
     },