Rayugacodes commited on
Commit
e835579
·
verified ·
1 Parent(s): 1dc42c1

Fix: restart runtime after pip install to resolve sympy/torch conflict

Browse files
Files changed (1) hide show
  1. KernelX_Training.ipynb +6 -1
KernelX_Training.ipynb CHANGED
@@ -45,7 +45,12 @@
45
  "execution_count": null,
46
  "metadata": {},
47
  "outputs": [],
48
- "source": "!pip install -q \"sympy>=1.13,<1.14\" torch \"transformers==4.46.3\" \"trl==0.15.2\" \"peft>=0.13,<0.15\" datasets \"accelerate>=0.34,<0.36\" huggingface_hub\nprint('Dependencies installed')"
 
 
 
 
 
49
  },
50
  {
51
  "cell_type": "code",
 
45
  "execution_count": null,
46
  "metadata": {},
47
  "outputs": [],
48
+ "source": "!pip install -q \"sympy>=1.13,<1.14\" \"torch>=2.1,<2.5\" \"transformers==4.46.3\" \"trl==0.15.2\" \"peft>=0.13,<0.15\" datasets \"accelerate>=0.34,<0.36\" huggingface_hub\n\n# Restart runtime to pick up new torch/sympy versions\nimport os\nos.kill(os.getpid(), 9)"
49
+ },
50
+ {
51
+ "cell_type": "markdown",
52
+ "source": "**After the cell above runs, the runtime will restart. This is expected.** Once it restarts, continue running from the next cell below (skip the install cell).",
53
+ "metadata": {}
54
  },
55
  {
56
  "cell_type": "code",