docker fix, rm requirements.txt

Dockerfile

@@ -1,6 +1,6 @@
 FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
 
-WORKDIR /code
+WORKDIR /code/MMAudio
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -27,35 +27,35 @@ RUN pip3 install --no-cache-dir numpy==1.24.3
 
 # Install PyTorch and related packages first (as recommended in README)
 RUN pip3 install --no-cache-dir \
-    torch==2.6.0 \
-    torchvision==0.21.0 \
-    torchaudio==2.6.0 \
-    --index-url https://download.pytorch.org/whl/cu118
+    torch \
+    torchvision \
+    torchaudio \
+    --index-url https://download.pytorch.org/whl/cu118 --upgrade
 
 # Install core dependencies
-RUN pip3 install --no-cache-dir \
-    colorlog==6.8.2 \
-    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
+RUN pip3 install --no-cache-dir -e .
+    # colorlog==6.8.2 \
+    # 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
 
 # Clone MMAudio
-RUN git clone https://github.com/hkchengrex/MMAudio.git
+# RUN git clone https://github.com/hkchengrex/MMAudio.git
 
 # Set working directory to MMAudio
-WORKDIR /code/MMAudio
+# WORKDIR /code/MMAudio
 
 # Install other dependencies
-RUN pip3 install --no-cache-dir -r requirements.txt
+# RUN pip3 install --no-cache-dir -r requirements.txt
 
 # Install MMAudio last (as recommended in README)
-RUN pip3 install -e .
+# RUN pip3 install -e .
 
 # Create output directory
 RUN mkdir -p output/gradio && chmod 777 output/gradio

requirements.txt

@@ -1,25 +1,25 @@
-numpy==1.24.3
-torch==2.6.0
-torchvision==0.21.0
-torchaudio==2.6.0
-torchdiffeq==0.2.3
-omegaconf>=2.3.0
-huggingface_hub>=0.26.1
-Pillow>=9.5
-opencv-python-headless>=4.8
-gradio>=4.19.2
-einops>=0.6
-open_clip_torch>=2.29.0
-av>=14.0.1
-transformers>=4.36.2
-ffmpeg-python>=0.2.0
-moviepy>=1.0.3
-python-multipart>=0.0.9
-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
+# numpy==1.24.3
+# torch==2.6.0
+# torchvision==0.21.0
+# torchaudio==2.6.0
+# torchdiffeq==0.2.3
+# omegaconf>=2.3.0
+# huggingface_hub>=0.26.1
+# Pillow>=9.5
+# opencv-python-headless>=4.8
+# gradio>=4.19.2
+# einops>=0.6
+# open_clip_torch>=2.29.0
+# av>=14.0.1
+# transformers>=4.36.2
+# ffmpeg-python>=0.2.0
+# moviepy>=1.0.3
+# python-multipart>=0.0.9
+# 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