Fix dependencies

.gitignore

@@ -0,0 +1,8 @@
+models/dwpose/
+models/musetalk/
+models/musetalkV15/
+models/sd-vae/
+models/syncnet/
+models/whisper/
+
+__pycache__/

install.sh

@@ -19,15 +19,12 @@ ln -s ~/.cache/huggingface/hub/models--ByteDance--LatentSync/snapshots/*/* ~/app
 mkdir ~/app/models/whisper
 ln -s ~/.cache/huggingface/hub/models--openai--whisper-tiny/snapshots/*/* ~/app/models/whisper/
 
-# Hugging face only allows to install torch atleast version 2.1.0, so we reinstalling it at runtime
-#pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
-
 pip install --no-cache-dir -U openmim
 
 # Mim uses ~/.cache/mim by default for cache, but hugging face only allows to use /home/user/app, so we use little hack
 export HOME=/home/user/app
 
-#mim install mmengine
-#mim install "mmcv==2.1.0"
-#mim install "mmdet==3.1.0"
-#mim install "mmpose==1.1.0"
+mim install mmengine
+mim install "mmcv==2.1.0"
+mim install "mmdet==3.2.0"
+mim install "mmpose==1.2.0"

requirements.txt

@@ -1,4 +1,4 @@
---extra-index-url https://download.pytorch.org/whl/cu118
+--extra-index-url https://download.pytorch.org/whl/cu121
 
 diffusers==0.30.2
 accelerate==0.28.0
@@ -21,8 +21,6 @@ omegaconf
 ffmpeg-python
 moviepy
 
-#torch~=2.1.0
-#torchvision~=0.16.0
-#torchaudio~=2.1.0
-
-#openmim
+torch==2.1.0
+torchvision==0.16.0
+torchaudio==2.1.0