Update requirements.txt

requirements.txt

@@ -1,33 +1,34 @@
---find-links https://download.pytorch.org/whl/torch_stable.html
-
-accelerate==0.28.0
-audio-separator==0.17.2
-av==12.1.0
-bitsandbytes==0.43.1
-decord==0.6.0
-diffusers==0.27.2
-einops==0.8.0
-insightface==0.7.3
-librosa==0.10.2.post1
-mediapipe[vision]==0.10.14
-mlflow==2.13.1
-moviepy==1.0.3
-numpy==1.26.4
-omegaconf==2.3.0
-onnx2torch==1.5.14
-onnx==1.16.1
-onnxruntime-gpu==1.18.0
-opencv-contrib-python==4.9.0.80
-opencv-python-headless==4.9.0.80
-opencv-python==4.9.0.80
-pillow==10.3.0
-setuptools==70.0.0
-torch==2.2.2+cu121
-torchvision==0.17.2+cu121
-tqdm==4.66.4
-transformers==4.39.2
-xformers==0.0.25.post1
-isort==5.13.2
-pylint==3.2.2
-pre-commit==3.7.1
-gradio==4.36.1
+--find-links https://download.pytorch.org/whl/torch_stable.html
+
+accelerate==0.28.0
+audio-separator==0.17.2
+av==12.1.0
+bitsandbytes==0.43.1
+decord==0.6.0
+diffusers==0.27.2
+einops==0.8.0
+insightface==0.7.3
+librosa==0.10.2.post1
+mediapipe[vision]==0.10.14
+mlflow==2.13.1
+moviepy==1.0.3
+numpy==1.26.4
+omegaconf==2.3.0
+onnx2torch==1.5.14
+onnx==1.16.1
+onnxruntime-gpu==1.18.0
+opencv-contrib-python==4.9.0.80
+opencv-python-headless==4.9.0.80
+opencv-python==4.9.0.80
+pillow==10.3.0
+setuptools==70.0.0
+tqdm==4.66.4
+transformers==4.39.2
+isort==5.13.2
+pylint==3.2.2
+pre-commit==3.7.1
+gradio==4.36.1
+torch==2.2.2+cu121 --index-url https://download.pytorch.org/whl/cu121
+torchvision==0.17.2+cu121 --index-url https://download.pytorch.org/whl/cu121
+torchaudio==2.2.2+cu121 --index-url https://download.pytorch.org/whl/cu121
+xformers==0.0.25.post1