Update requirements.txt

requirements.txt

@@ -1,44 +1,52 @@
-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/[email protected]#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
+pytest
 matplotlib
 jupyter
 ipython
-rich
-pytest
 ipdb
 gpustat
 scikit-learn
 spacy
 munch
+einops
+opt_einsum
 fvcore
+gsutil
+cmake
+pykeops
+zstandard
 psutil
+h5py
 twine
 gdown
+hydra-core==1.3.1
+hydra-colorlog==1.2.0
+hydra-optuna-sweeper==1.2.0
+pyrootutils
+rich
+transformers==4.45.2
+datasets==3.0.1
+pytorch-lightning==2.2.1
+triton==2.3.1
+wandb==0.16.3
+controlnet_aux==0.0.9
+timm==0.6.7
+torchmetrics==1.3.2
+git+https://github.com/mlcommons/[email protected]
+accelerate==0.34.2
+diffusers==0.30.3
+deepspeed==0.15.2
+sentencepiece==0.1.99
+pillow==10.2.0
+torch==2.3.1
+torchvision==0.18.1
+torchaudio==2.3.1
+flash-attn==2.6.3
+git+https://github.com/Dao-AILab/[email protected]#subdirectory=csrc/fused_dense_lib
 jaxtyping
-gradio
+mediapipe
+gradio
+git+https://github.com/facebookresearch/pytorch3d.git
+opencv-python==4.5.5.64
+opencv-python-headless==4.5.5.64
+huggingface_hub==0.24
+numpy==1.24.4