FIX Use correct model in image clf notebook (#1624)

huggingface · Apr 5, 2024 · 2e821c1 · 2e821c1
1 parent 8452d71
commit 2e821c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/image_classification/image_classification_peft_lora.ipynb b/examples/image_classification/image_classification_peft_lora.ipynb
@@ -1743,7 +1743,7 @@
    ],
    "source": [
     "trainer = Trainer(\n",
-    "    model,\n",
+    "    lora_model,\n",
     "    args,\n",
     "    train_dataset=train_ds,\n",
     "    eval_dataset=val_ds,\n",