requirements.txt

absl-py==2.0.0
accelerate==0.25.0
annotated-types==0.6.0
appdirs==1.4.4
bitsandbytes==0.41.3.post2
cachetools==5.3.2
certifi==2023.11.17
charset-normalizer==3.3.2
click==8.1.7
contourpy==1.2.0
cycler==0.12.1
deepspeed==0.12.6
diffusers==0.25.0
docker-pycreds==0.4.0
einops==0.7.0
filelock==3.9.0
fonttools==4.47.0
fsspec==2023.12.2
ftfy==6.1.3
gitdb==4.0.11
GitPython==3.1.41
google-auth==2.26.1
google-auth-oauthlib==1.2.0
grpcio==1.60.0
hjson==3.1.0
huggingface-hub==0.20.1
idna==3.6
importlib-metadata==7.0.1
Jinja2==3.1.2
kiwisolver==1.4.5
Markdown==3.5.1
MarkupSafe==2.1.3
matplotlib==3.8.2
mpmath==1.3.0
networkx==3.0
ninja==1.11.1.1
numpy==1.24.1
oauthlib==3.2.2
opencv-python==4.9.0.80
packaging==23.2
peft==0.7.1
pillow==10.2.0
protobuf==4.23.4
psutil==5.9.7
py-cpuinfo==9.0.0
pyasn1==0.5.1
pyasn1-modules==0.3.0
pydantic==2.5.3
pydantic_core==2.14.6
pynvml==11.5.0
pyparsing==3.1.1
python-dateutil==2.8.2
PyYAML==6.0.1
regex==2023.12.25
requests==2.31.0
requests-oauthlib==1.3.1
rsa==4.9
safetensors==0.4.1
scipy==1.11.4
sentry-sdk==1.39.2
setproctitle==1.3.3
six==1.16.0
smmap==5.0.1
sympy==1.12
tensorboard==2.15.1
tensorboard-data-server==0.7.2
tokenizers==0.15.0
torch==2.1.2+cu118
torchvision==0.16.2
tqdm==4.66.1
transformers==4.36.2
trimesh==4.0.8
triton==2.1.0
typing_extensions==4.9.0
urllib3==2.1.0
wandb==0.16.2
wcwidth==0.2.12
Werkzeug==3.0.1
xformers==0.0.23.post1+cu118
zipp==3.17.0