Upload EthicalHacking_LFM2.5_Ultimate_Colab.ipynb

EthicalHacking_LFM2.5_Ultimate_Colab.ipynb

@@ -66,7 +66,10 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## 3️⃣ Load LFM2.5-1.2B-Instruct in 4-bit via Unsloth"
+    "## 3️⃣ Load LFM2.5-1.2B-Instruct in 4-bit via Unsloth\n",
+    "\n",
+    "**⚠️ IMPORTANT:** We add `device_map={\"\": torch.cuda.current_device()}` to force the model onto the correct GPU.\n",
+    "Without this, `accelerate` may place the model on CPU and throw a `ValueError` during training on Kaggle/Colab."
    ]
   },
   {
@@ -97,6 +100,7 @@
     "    max_seq_length=MAX_SEQ_LENGTH,\n",
     "    dtype=None,\n",
     "    load_in_4bit=True,\n",
+    "    device_map={\"\": torch.cuda.current_device()},  # ← FORCE GPU: fixes Kaggle/Colab device placement bug\n",
     ")\n",
     "\n",
     "model = FastLanguageModel.get_peft_model(\n",