Create DOCKERFILE

DOCKERFILE

@@ -0,0 +1,55 @@
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
+ENV CONDA_DIR=/opt/conda
+ENV PATH=$CONDA_DIR/bin:$PATH
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    git-lfs \
+    wget \
+    curl \
+    ca-certificates \
+    build-essential \
+    libsndfile1 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Miniconda
+RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
+    /bin/bash ~/miniconda.sh -b -p $CONDA_DIR && \
+    rm ~/miniconda.sh && \
+    conda clean -tipsy
+
+# Create conda environment
+RUN conda create -n yue python=3.8 -y
+SHELL ["conda", "run", "-n", "yue", "/bin/bash", "-c"]
+
+# Install PyTorch with CUDA support
+RUN conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia -y
+
+# Install flash-attention
+RUN pip install flash-attn --no-build-isolation
+
+# Clone YuE repository and install dependencies
+WORKDIR /app
+RUN git lfs install && \
+    git clone https://github.com/multimodal-art-projection/YuE.git && \
+    cd YuE && \
+    pip install -r requirements.txt
+
+# Clone xcodec_mini_infer
+WORKDIR /app/YuE/inference
+RUN git clone https://huggingface.co/m-a-p/xcodec_mini_infer
+
+# Create output directory
+RUN mkdir -p /app/output
+
+# Copy entry script and wrapper
+COPY run.sh /app/
+COPY wrapper.py /app/YuE/inference/
+RUN chmod +x /app/run.sh
+
+ENTRYPOINT ["/app/run.sh"]