Skip to content

GitLab

Unverified Commit 6a611955 authored by yanxiyue's avatar yanxiyue Committed by GitHub
Browse files

lock torch version to 2.0.1 (#1290)

parent b95ee898
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
pyproject.toml
View file @ 6a611955
...@@ -3,7 +3,7 @@ requires = [ ...@@ -3,7 +3,7 @@ requires = [
"ninja", "ninja",
"packaging", "packaging",
"setuptools", "setuptools",
"torch >= 2.0.0", "torch == 2.0.1",
"wheel", "wheel",
] ]
build-backend = "setuptools.build_meta" build-backend = "setuptools.build_meta"
requirements.txt
View file @ 6a611955
...@@ -5,9 +5,9 @@ pandas # Required for Ray data. ...@@ -5,9 +5,9 @@ pandas # Required for Ray data.
pyarrow # Required for Ray data. pyarrow # Required for Ray data.
sentencepiece # Required for LLaMA tokenizer. sentencepiece # Required for LLaMA tokenizer.
numpy numpy
torch >= 2.0.0 torch == 2.0.1
transformers >= 4.33.1 # Required for Code Llama. transformers >= 4.33.1 # Required for Code Llama.
xformers >= 0.0.22 xformers == 0.0.22
fastapi fastapi
uvicorn[standard] uvicorn[standard]
pydantic < 2 # Required for OpenAI server. pydantic < 2 # Required for OpenAI server.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment