Fix: let pip resolve all versions, uninstall old transformers, auto-restart if needed

KernelX_Training.ipynb

@@ -45,7 +45,7 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": "# Clean install: remove old transformers first, then install compatible versions\n!pip uninstall -y transformers trl peft accelerate -q\n!pip install -q \"transformers==4.48.3\" \"trl==0.15.2\" \"peft==0.14.0\" \"datasets\" \"accelerate==0.36.2\" \"huggingface_hub\"\n\n# Must restart runtime after reinstalling transformers\nimport os\nos.kill(os.getpid(), 9)"
+   "source": "# Step 1: Remove old conflicting transformers\n!pip uninstall -y transformers -q 2>/dev/null\n\n# Step 2: Install training libraries (let pip resolve versions automatically)\n!pip install -q trl peft datasets accelerate huggingface_hub\n\n# Step 3: Verify everything imports\ntry:\n    import transformers, trl, peft, torch\n    print(f'torch={torch.__version__} transformers={transformers.__version__} trl={trl.__version__} peft={peft.__version__}')\n    print(f'CUDA: {torch.cuda.is_available()}')\n    from transformers import AutoModelForCausalLM, AutoTokenizer\n    from trl import SFTTrainer, SFTConfig\n    print('All imports OK - no restart needed!')\nexcept Exception as e:\n    print(f'Import failed: {e}')\n    print('Restarting runtime...')\n    import os\n    os.kill(os.getpid(), 9)"
   },
   {
    "cell_type": "markdown",