Update requirements.txt

requirements.txt

@@ -1,6 +1,64 @@
-accelerate
-diffusers
-invisible_watermark
-torch
-transformers
-xformers
+accelerate==0.32.1
+bitsandbytes==0.43.1
+certifi==2024.7.4
+charset-normalizer==3.3.2
+click==8.1.7
+cramjam==2.8.3
+diffusers @ git+https://github.com/huggingface/diffusers@b8cf84a3f902550937255c5b28b39827ba52beb6
+docker-pycreds==0.4.0
+fastparquet==2024.5.0
+filelock==3.15.4
+fsspec==2024.6.1
+gitdb==4.0.11
+GitPython==3.1.43
+huggingface-hub==0.23.4
+idna==3.7
+importlib_metadata==8.0.0
+Jinja2==3.1.4
+MarkupSafe==2.1.5
+mpmath==1.3.0
+networkx==3.3
+numpy==1.26.4
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.20.5
+nvidia-nvjitlink-cu12==12.5.82
+nvidia-nvtx-cu12==12.1.105
+packaging==24.1
+pandas==2.2.2
+peft==0.11.1
+pillow==10.4.0
+platformdirs==4.2.2
+protobuf==5.27.2
+psutil==6.0.0
+pyarrow==16.1.0
+python-dateutil==2.9.0.post0
+pytz==2024.1
+PyYAML==6.0.1
+regex==2024.5.15
+requests==2.32.3
+safetensors==0.4.3
+sentencepiece==0.1.96
+sentry-sdk==2.9.0
+setproctitle==1.3.3
+six==1.16.0
+smmap==5.0.1
+sympy==1.13.0
+tokenizers==0.19.1
+torch==2.3.1
+torchvision==0.18.1
+tqdm==4.66.4
+transformers==4.42.4
+triton==2.3.1
+typing_extensions==4.12.2
+tzdata==2024.1
+urllib3==2.2.2
+wandb==0.17.4
+zipp==3.19.2