tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

lock pytorch (#1247) [skip ci]

Browse Source
This commit is contained in:
JohanWork
2024-02-06 13:48:26 +01:00
committed by GitHub
parent 13eea21f9b
commit 1c7ed26785
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/colab-notebooks/colab-axolotl-example.ipynb
View File

@@ -43,6 +43,7 @@
}, },
"outputs": [], "outputs": [],
"source": [ "source": [
"!pip install torch==\"2.1.2\"\n",
"!pip install -e git+https://github.com/OpenAccess-AI-Collective/axolotl#egg=axolotl\n", "!pip install -e git+https://github.com/OpenAccess-AI-Collective/axolotl#egg=axolotl\n",
"!pip install flash-attn==\"2.5.0\"\n", "!pip install flash-attn==\"2.5.0\"\n",
"!pip install deepspeed==\"0.13.1\"" "!pip install deepspeed==\"0.13.1\""