dependencies

Dockerfile

@@ -12,6 +12,7 @@ RUN apt-get update && apt-get install -y \
     libsm6 \
     libxext6 \
     curl \
+    libsndfile1 \
     && rm -rf /var/lib/apt/lists/*
 
 # Ensure we're using Python 3.10
@@ -31,10 +32,19 @@ RUN pip3 install --no-cache-dir \
     torchaudio==2.1.2 \
     --index-url https://download.pytorch.org/whl/cu118
 
-# Install colorlog and torchdiffeq
+# Install core dependencies
 RUN pip3 install --no-cache-dir \
     colorlog==6.8.2 \
-    torchdiffeq==0.2.3
+    torchdiffeq==0.2.3 \
+    omegaconf>=2.3.0 \
+    pandas>=2.0.0 \
+    tensordict>=0.2.0 \
+    hydra-core>=1.3.2 \
+    tqdm>=4.65.0 \
+    librosa>=0.10.1 \
+    timm>=0.9.12 \
+    requests>=2.31.0 \
+    torio>=0.3.1
 
 # Clone MMAudio
 RUN git clone https://github.com/hkchengrex/MMAudio.git

requirements.txt

@@ -3,6 +3,7 @@ torch==2.1.2
 torchvision==0.16.2
 torchaudio==2.1.2
 torchdiffeq==0.2.3
+omegaconf>=2.3.0
 huggingface_hub>=0.26.1
 Pillow>=9.5
 opencv-python-headless>=4.8
@@ -14,4 +15,12 @@ transformers>=4.36.2
 ffmpeg-python>=0.2.0
 moviepy>=1.0.3
 python-multipart>=0.0.9
-colorlog==6.8.2
+colorlog==6.8.2
+pandas>=2.0.0
+tensordict>=0.2.0
+hydra-core>=1.3.2
+tqdm>=4.65.0
+librosa>=0.10.1
+timm>=0.9.12
+requests>=2.31.0
+torio>=0.3.1