Update requirements.txt

requirements.txt

@@ -1,6 +1,44 @@
-accelerate
-diffusers
-invisible_watermark
-torch
-transformers
-xformers
+transformers==4.45.2
+pytorch-lightning==2.2.1
+accelerate==0.34.2
+deepspeed==0.15.2
+triton==2.3.1
+diffusers==0.30.3
+controlnet_aux==0.0.9
+opencv-python==4.5.5.64
+opencv-python-headless==4.5.5.64
+pillow==10.2.0
+mediapipe
+flash-attn==2.6.3
+flash-attention-fused-dense @ git+https://github.com/Dao-AILab/flash-attention@v2.6.3#subdirectory=csrc/fused_dense_lib
+datasets==3.0.1
+huggingface_hub==0.24
+sentencepiece==0.1.99
+timm==0.6.7
+torchmetrics==1.3.2
+einops
+opt_einsum
+pykeops
+zstandard
+h5py
+wandb==0.16.3
+hydra-core==1.3.1
+hydra-colorlog==1.2.0
+hydra-optuna-sweeper==1.2.0
+pyrootutils
+matplotlib
+jupyter
+ipython
+rich
+pytest
+ipdb
+gpustat
+scikit-learn
+spacy
+munch
+fvcore
+psutil
+twine
+gdown
+jaxtyping
+gradio