Fix: keep Colab default torch, only install sympy fix + training libs
Browse files- KernelX_Training.ipynb +1 -1
KernelX_Training.ipynb
CHANGED
|
@@ -45,7 +45,7 @@
|
|
| 45 |
"execution_count": null,
|
| 46 |
"metadata": {},
|
| 47 |
"outputs": [],
|
| 48 |
-
"source": "!pip install -q \"sympy>=1.13,<1.14\" \"
|
| 49 |
},
|
| 50 |
{
|
| 51 |
"cell_type": "markdown",
|
|
|
|
| 45 |
"execution_count": null,
|
| 46 |
"metadata": {},
|
| 47 |
"outputs": [],
|
| 48 |
+
"source": "!pip install -q \"sympy>=1.13,<1.14\" \"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 sympy version (torch stays as Colab's default)\nimport os\nos.kill(os.getpid(), 9)"
|
| 49 |
},
|
| 50 |
{
|
| 51 |
"cell_type": "markdown",
|