wassname/latent-gemma
1
0
Fork 0
mirror of https://github.com/wassname/latent-gemma.git synced 2026-09-11 12:30:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

update notebook

Browse Source
This commit is contained in:
Victor Umesiobi
2025-01-03 07:15:50 +00:00
parent 36ee48699a
commit b0260a204a
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fine_tuning_gemma2.ipynb
+4 -4
View File
@@ -2968,14 +2968,14 @@
},
{
"cell_type": "code",
"execution_count": 4,
"execution_count": 15,
"metadata": {},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"Loading checkpoint shards: 100%|██████████| 2/2 [00:01<00:00, 1.26it/s]\n"
"Loading checkpoint shards: 100%|██████████| 2/2 [00:05<00:00, 2.64s/it]\n"
]
}
],
@@ -2983,7 +2983,7 @@
"from transformers import AutoTokenizer, AutoConfig, AutoModelForCausalLM\n",
"from latent_gemma import LatentReasoningGemmaForCausalLM\n",
"\n",
"model_path = \"/home/featurize/.cache/kagglehub/models/victorumesiobi/gemma-2-japanese-english-reasoning/transformers/1/1/transformers\" # Replace with the path to which your model was downloaded too\n",
"model_path = \"/home/featurize/.cache/kagglehub/models/victorumesiobi/gemma-2-japanese-english-reasoning/transformers/1/2\" # Replace with the path to which your model was downloaded too\n",
"\n",
"# Load model and tokenizer\n",
"tokenizer = AutoTokenizer.from_pretrained(model_path)\n",
@@ -3044,7 +3044,7 @@
},
{
"cell_type": "code",
"execution_count": 13,
"execution_count": 14,
"metadata": {},
"outputs": [
{