lock pytorch (#1247) [skip ci]

2024-02-06 13:48:26 +01:00
parent 13eea21f9b
commit 1c7ed26785
1 changed files with 1 additions and 0 deletions
--- a/examples/colab-notebooks/colab-axolotl-example.ipynb
+++ b/examples/colab-notebooks/colab-axolotl-example.ipynb
@@ -43,6 +43,7 @@
      },
      "outputs": [],
      "source": [
+        "!pip install torch==\"2.1.2\"\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 deepspeed==\"0.13.1\""