From 1c7ed267851afd3967cde5e34cbe0d0cdade8b66 Mon Sep 17 00:00:00 2001
From: JohanWork <39947546+JohanWork@users.noreply.github.com>
Date: Tue, 6 Feb 2024 13:48:26 +0100
Subject: [PATCH] lock pytorch (#1247) [skip ci]

---
 examples/colab-notebooks/colab-axolotl-example.ipynb | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/colab-notebooks/colab-axolotl-example.ipynb b/examples/colab-notebooks/colab-axolotl-example.ipynb
index f9a7f65da..44b48ddb8 100644
--- 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\""