Update requirements.txt
Browse files- requirements.txt +17 -22
requirements.txt
CHANGED
|
@@ -1,19 +1,20 @@
|
|
| 1 |
-
#
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 6 |
|
| 7 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
diffusers==0.33.1
|
| 9 |
transformers==4.52.3
|
| 10 |
-
huggingface-hub
|
| 11 |
accelerate==0.26.1
|
| 12 |
peft==0.5.0
|
| 13 |
-
xformers==0.0.29.post1
|
| 14 |
-
triton==3.1.0
|
| 15 |
|
| 16 |
-
# MMAudio deps
|
| 17 |
python-dotenv
|
| 18 |
cython
|
| 19 |
gitpython>=3.1
|
|
@@ -31,24 +32,18 @@ tensordict
|
|
| 31 |
colorlog
|
| 32 |
open_clip_torch
|
| 33 |
soundfile
|
| 34 |
-
torchaudio
|
| 35 |
|
| 36 |
-
# LatentSync & video
|
| 37 |
imageio==2.27.0
|
| 38 |
decord==0.6.0
|
| 39 |
opencv-python==4.9.0.80
|
| 40 |
mediapipe==0.10.11
|
| 41 |
av==11.0.0
|
| 42 |
-
torch-fidelity==0.3.0
|
| 43 |
-
torchmetrics==1.3.1
|
| 44 |
-
python_speech_features==0.6
|
| 45 |
-
librosa==0.10.1
|
| 46 |
-
scenedetect==0.6.1
|
| 47 |
ffmpeg-python==0.2.0
|
| 48 |
-
lpips==0.1.4
|
| 49 |
face-alignment==1.4.1
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
numpy==1.24.4
|
| 53 |
-
pydub==0.25.1
|
| 54 |
moviepy==1.0.3
|
|
|
|
|
|
|
|
|
| 1 |
+
# ==== PyTorch + CUDA 11.8 cho Colab Free (T4) ====
|
| 2 |
+
torch==2.5.1+cu118
|
| 3 |
+
torchvision==0.20.1+cu118
|
| 4 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|
|
|
|
| 5 |
|
| 6 |
+
# ==== Giao diện & cấu hình ====
|
| 7 |
+
gradio>=3.35.0
|
| 8 |
+
omegaconf>=2.3.0
|
| 9 |
+
|
| 10 |
+
# ==== HuggingFace & diffusion ====
|
| 11 |
diffusers==0.33.1
|
| 12 |
transformers==4.52.3
|
| 13 |
+
huggingface-hub>=0.27.0,<1.0.0
|
| 14 |
accelerate==0.26.1
|
| 15 |
peft==0.5.0
|
|
|
|
|
|
|
| 16 |
|
| 17 |
+
# ==== MMAudio (Long_Tieng) deps ====
|
| 18 |
python-dotenv
|
| 19 |
cython
|
| 20 |
gitpython>=3.1
|
|
|
|
| 32 |
colorlog
|
| 33 |
open_clip_torch
|
| 34 |
soundfile
|
| 35 |
+
torchaudio>=2.0.2
|
| 36 |
|
| 37 |
+
# ==== LatentSync & video xử lý deps ====
|
| 38 |
imageio==2.27.0
|
| 39 |
decord==0.6.0
|
| 40 |
opencv-python==4.9.0.80
|
| 41 |
mediapipe==0.10.11
|
| 42 |
av==11.0.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
ffmpeg-python==0.2.0
|
|
|
|
| 44 |
face-alignment==1.4.1
|
| 45 |
+
librosa==0.10.1
|
| 46 |
+
python_speech_features==0.6
|
|
|
|
|
|
|
| 47 |
moviepy==1.0.3
|
| 48 |
+
numpy>=1.22.0
|
| 49 |
+
pandas>=2.0.0
|