Upload New File

54becb5c · wanglch · 428df697 · 54becb5c
Commit 54becb5c authored Aug 26, 2024 by wanglch
Hide whitespace changes
Inline Side-by-side

Showing with 17 additions and 0 deletions

docker/requirements.txt docker/requirements.txt +17 -0

No files found.
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
+peft>=0.10.0
+deepspeed>=0.14.2
+mpi4py>=3.1.4
+tensorboard>=2.16.2
+xformers
+transformers==4.40.2
+huggingface-hub>=0.23.0
+pillow
+chainlit>=1.0
+pydantic>=2.7.1
+timm>=0.9.16
+openai>=1.30.1
+loguru>=0.7.2
+pydantic>=2.7.1
+einops
+sse-starlette
+bitsandbytes