Spaces:
Runtime error
Runtime error
File size: 1,039 Bytes
73ed896 f47eaa6 73ed896 f47eaa6 73ed896 f47eaa6 73ed896 f47eaa6 73ed896 f47eaa6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
WORKDIR /code
# Install system dependencies
RUN apt-get update && apt-get install -y \
python3.9 \
python3-pip \
git \
ffmpeg \
libsm6 \
libxext6 \
&& rm -rf /var/lib/apt/lists/*
# Clone MMAudio and install dependencies
RUN git clone https://github.com/hkchengrex/MMAudio.git && \
cd MMAudio && \
# Install PyTorch first as specified in README
pip3 install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118 && \
# Install additional dependencies
pip3 install -r requirements.txt && \
# Install MMAudio
pip3 install -e .
# Set working directory to MMAudio
WORKDIR /code/MMAudio
# Create output directory
RUN mkdir -p output/gradio && chmod 777 output/gradio
# Set environment variables for Hugging Face Spaces
ENV PYTHONUNBUFFERED=1
ENV GRADIO_SERVER_NAME=0.0.0.0
ENV GRADIO_SERVER_PORT=7860
# Expose Gradio port
EXPOSE 7860
# Run the Gradio demo
CMD ["python3", "gradio_demo.py"] |