Skip to content

Commit 6a61195

Browse files
authored
lock torch version to 2.0.1 (#1290)
1 parent b95ee89 commit 6a61195

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ requires = [
33
"ninja",
44
"packaging",
55
"setuptools",
6-
"torch >= 2.0.0",
6+
"torch == 2.0.1",
77
"wheel",
88
]
99
build-backend = "setuptools.build_meta"

requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@ pandas # Required for Ray data.
55
pyarrow # Required for Ray data.
66
sentencepiece # Required for LLaMA tokenizer.
77
numpy
8-
torch >= 2.0.0
8+
torch == 2.0.1
99
transformers >= 4.33.1 # Required for Code Llama.
10-
xformers >= 0.0.22
10+
xformers == 0.0.22
1111
fastapi
1212
uvicorn[standard]
1313
pydantic < 2 # Required for OpenAI server.

0 commit comments

Comments
 (0)