From 6a6119554cee13bb4ef5018155be5530d6aaca26 Mon Sep 17 00:00:00 2001
From: yanxiyue <alex.yueye@gmail.com>
Date: Wed, 11 Oct 2023 00:21:57 +0800
Subject: [PATCH] lock torch version to 2.0.1 (#1290)

---
 pyproject.toml   | 2 +-
 requirements.txt | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 264566404f22..360e023a05a0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,7 +3,7 @@ requires = [
     "ninja",
     "packaging",
     "setuptools",
-    "torch >= 2.0.0",
+    "torch == 2.0.1",
     "wheel",
 ]
 build-backend = "setuptools.build_meta"
diff --git a/requirements.txt b/requirements.txt
index e7491957fb8f..dea25ac6d58c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,9 +5,9 @@ pandas  # Required for Ray data.
 pyarrow  # Required for Ray data.
 sentencepiece  # Required for LLaMA tokenizer.
 numpy
-torch >= 2.0.0
+torch == 2.0.1
 transformers >= 4.33.1  # Required for Code Llama.
-xformers >= 0.0.22
+xformers == 0.0.22
 fastapi
 uvicorn[standard]
 pydantic < 2  # Required for OpenAI server.