JohanWork
commited on
Commit
•
1c7ed26
1
Parent(s):
13eea21
lock pytorch (#1247) [skip ci]
Browse files
examples/colab-notebooks/colab-axolotl-example.ipynb
CHANGED
|
@@ -43,6 +43,7 @@
|
|
| 43 |
},
|
| 44 |
"outputs": [],
|
| 45 |
"source": [
|
|
|
|
| 46 |
"!pip install -e git+https://github.com/OpenAccess-AI-Collective/axolotl#egg=axolotl\n",
|
| 47 |
"!pip install flash-attn==\"2.5.0\"\n",
|
| 48 |
"!pip install deepspeed==\"0.13.1\""
|
|
|
|
| 43 |
},
|
| 44 |
"outputs": [],
|
| 45 |
"source": [
|
| 46 |
+
"!pip install torch==\"2.1.2\"\n",
|
| 47 |
"!pip install -e git+https://github.com/OpenAccess-AI-Collective/axolotl#egg=axolotl\n",
|
| 48 |
"!pip install flash-attn==\"2.5.0\"\n",
|
| 49 |
"!pip install deepspeed==\"0.13.1\""
|