Update Dockerfile

Dockerfile

@@ -43,28 +43,26 @@ RUN conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c
 # Install flash-attention
 RUN pip install flash-attn --no-build-isolation
 
-# Set up app directory
+# First clone repositories and create directories
 WORKDIR $HOME/app
 
-# Clone YuE repository and install dependencies
 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 $HOME/app/YuE/inference
 RUN git clone https://huggingface.co/m-a-p/xcodec_mini_infer
 
 # Create output directory
 RUN mkdir -p $HOME/app/output
 
-# Copy your files
+# Copy only the necessary additional files
 COPY --chown=user:user run.sh $HOME/app/
 COPY --chown=user:user wrapper.py $HOME/app/YuE/inference/
 RUN chmod +x $HOME/app/run.sh
 
-# Set final working directory to where run.sh is located
+# Set final working directory
 WORKDIR $HOME/app
 
 ENTRYPOINT ["./run.sh"]