Fix: restart runtime after pip install to resolve sympy/torch conflict
Browse files- 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\
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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",
|