Update requirements.txt

requirements.txt

@@ -1,42 +1,58 @@
-torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 xformers==0.0.28.post3 --index-url https://download.pytorch.org/whl/cu124
+# Torch and related libraries
+torch==2.5.1+cu124
+torchvision==0.20.1+cu124
+torchaudio==2.5.1+cu124
+xformers==0.0.28.post3 --index-url https://download.pytorch.org/whl/cu124
 torchao --index-url https://download.pytorch.org/whl/nightly/cu124
---no-deps facenet_pytorch==2.6.0
+
+# Facenet and media processing
+facenet_pytorch==2.6.0
 mediapipe
+
+# Transformers, diffusers, and machine learning
 transformers>=4.46.3
 diffusers==0.31.0
 torchmetrics
 torchtyping
-tqdm
-einops==0.8.0
-omegaconf==2.3.0
-opencv-python
-av==13.1.0
-gradio
 accelerate==1.1.1
-clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a
-decord==0.6.0
-diffusers==0.31.0
-einops==0.8.0
-gradio_client==1.4.3
+
+# Image and video processing
+opencv-python
+opencv-contrib-python==4.10.0.84
+Pillow>=10.4.0
 imageio==2.36.0
 imageio-ffmpeg==0.5.1
+av==13.1.0
+decord==0.6.0
+moviepy==1.0.3
+scenedetect
+
+# NumPy, SciPy, and related libraries
 numpy==1.26.4
-onnxruntime-gpu==1.20.1
-open-clip-torch==2.29.0
-opencv-contrib-python==4.10.0.84
-Pillow>=10.4.0
 scikit-image==0.24.0
 scikit-learn==1.5.2
 scipy==1.14.1
-torchdiffeq==0.2.5
-torchsde==0.2.6
+
+# Audio processing
+soundfile
+ffmpeg-python
+
+# Miscellaneous libraries
+einops==0.8.0
+omegaconf==2.3.0
 mlflow==2.18.0
 controlnet-aux==0.0.9
-ffmpeg-python
-soundfile
-mediapipe
-decord
 IPython
-scenedetect
-moviepy==1.0.3
+gradio
+gradio_client==1.4.3
 huggingface_hub==0.26.2
+torchdiffeq==0.2.5
+torchsde==0.2.6
+open-clip-torch==2.29.0
+
+# CLIP library from GitHub
+clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a
+
+# ONNX Runtime for GPU
+onnxruntime-gpu==1.20.1
+