Rayugacodes commited on
Commit
73d8be2
·
verified ·
1 Parent(s): 6de2213

Fix: uninstall old transformers first, pin exact versions, restart runtime

Browse files
Files changed (1) hide show
  1. KernelX_Training.ipynb +1 -1
KernelX_Training.ipynb CHANGED
@@ -45,7 +45,7 @@
45
  "execution_count": null,
46
  "metadata": {},
47
  "outputs": [],
48
- "source": "# First reinstall torchvision to match Colab's torch, then install training libs\n!pip install -q --no-deps torchvision torchaudio\n!pip install -q \"transformers>=4.47,<4.50\" \"trl>=0.14,<0.18\" \"peft>=0.13\" datasets accelerate huggingface_hub\n\n# Verify imports work before proceeding\nimport transformers, trl, peft\nprint(f'transformers={transformers.__version__}, trl={trl.__version__}, peft={peft.__version__}')\nprint('All imports OK')"
49
  },
50
  {
51
  "cell_type": "markdown",
 
45
  "execution_count": null,
46
  "metadata": {},
47
  "outputs": [],
48
+ "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)"
49
  },
50
  {
51
  "cell_type": "markdown",