[Minor] Fix xformers version (#2158)

This commit is contained in:
Woosuk Kwon 2023-12-17 02:28:02 -08:00 committed by GitHub
parent 30fb0956df
commit c9fadda543
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -7,7 +7,7 @@ sentencepiece # Required for LLaMA tokenizer.
numpy
torch == 2.1.2
transformers >= 4.36.0 # Required for Mixtral.
xformers == 0.0.23 # Required for CUDA 12.1.
xformers == 0.0.23.post1 # Required for CUDA 12.1.
fastapi
uvicorn[standard]
pydantic == 1.10.13 # Required for OpenAI server.