Fix colab example notebook (#1805) [skip ci]

This commit is contained in:
Sri Kainkaryam
2024-08-04 12:24:26 -05:00
committed by GitHub
parent 78b42a3fe1
commit 203816f7b4

View File

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