Update Dockerfile

Dockerfile

@@ -47,17 +47,22 @@ RUN pip install flash-attn --no-build-isolation
 # First clone repositories and create directories
 WORKDIR $HOME/app
 
+# Clone YuE repository and install dependencies with proper LFS handling
 RUN git lfs install && \
-    git clone --recursive https://github.com/multimodal-art-projection/YuE.git && \
+    git clone https://github.com/multimodal-art-projection/YuE.git && \
     cd YuE && \
+    git lfs fetch && \
     git lfs pull && \
-    cd inference && \
     git lfs checkout && \
-    cd .. && \
     pip install -r requirements.txt
-    
+
+# Go to inference directory and clone the required model
 WORKDIR $HOME/app/YuE/inference
-RUN git clone https://huggingface.co/m-a-p/xcodec_mini_infer
+
+# Clone xcodec_mini_infer with proper LFS handling
+RUN git clone https://huggingface.co/m-a-p/xcodec_mini_infer && \
+    cd xcodec_mini_infer && \
+    git lfs pull && \
 
 # Create output directory
 RUN mkdir -p $HOME/app/output