Upload Dockerfile with huggingface_hub

Dockerfile

@@ -23,16 +23,19 @@ RUN git lfs install
 RUN git clone --depth 1 https://github.com/comfyanonymous/ComfyUI.git
 
 # Clone custom nodes with shallow cloning to reduce download size and avoid rate limits
+# Note: PuLID is now handled separately via COPY instead of git clone
 RUN git clone --depth 1 https://github.com/cubiq/ComfyUI_IPAdapter_plus.git /app/ComfyUI/custom_nodes/IPAdapter_plus || echo "Failed to clone IPAdapter_plus but continuing" && \
     git clone --depth 1 https://github.com/Kosinkadink/ComfyUI-KJNodes.git /app/ComfyUI/custom_nodes/KJNodes || echo "Failed to clone KJNodes but continuing" && \
     git clone --depth 1 https://github.com/SSitu/ComfyUI_UltimateSDUpscale.git /app/ComfyUI/custom_nodes/UltimateSDUpscale || echo "Failed to clone UltimateSDUpscale but continuing" && \
-    git clone --depth 1 https://github.com/ltdrdata/PuLID_ComfyUI.git /app/ComfyUI/custom_nodes/PuLID || echo "Failed to clone PuLID but continuing" && \
     git clone --depth 1 https://github.com/cg196/ComfyUI-UseEverywhere.git /app/ComfyUI/custom_nodes/UseEverywhere || echo "Failed to clone UseEverywhere but continuing" && \
     git clone --depth 1 https://github.com/melMass/comfy-mtb.git /app/ComfyUI/custom_nodes/mtb || echo "Failed to clone mtb but continuing" && \
     git clone --depth 1 https://github.com/LucianoCirino/ComfyUI_Masquerade.git /app/ComfyUI/custom_nodes/Masquerade || echo "Failed to clone Masquerade but continuing" && \
     git clone --depth 1 https://github.com/Fannovel16/comfyui-save-image-with-geninfo.git /app/ComfyUI/custom_nodes/SaveGenInfo || echo "Failed to clone SaveGenInfo but continuing" && \
     git clone --depth 1 https://github.com/jamesWalker55/comfyui-various.git /app/ComfyUI/custom_nodes/Various || echo "Failed to clone Various but continuing"
 
+# Create directories for PuLID custom node
+RUN mkdir -p /app/ComfyUI/custom_nodes/PuLID
+
 # Create necessary directories
 RUN mkdir -p /app/ComfyUI/models/checkpoints \
     /app/ComfyUI/models/controlnet \
@@ -42,11 +45,23 @@ RUN mkdir -p /app/ComfyUI/models/checkpoints \
     /app/ComfyUI/input \
     /app/ComfyUI/output \
     /app/workflows \
-    /tmp/comfyui_user
+    /tmp/comfyui_user \
+    /tmp/comfyui_input \
+    /tmp/comfyui_output \
+    /tmp/comfyui_temp \
+    /tmp/workflows \
+    /tmp/comfyui_models \
+    /tmp/comfyui_models/pulid
 
 # Set permissions for directories that need write access
 RUN chmod -R 777 /app/ComfyUI/output \
-    /tmp/comfyui_user
+    /tmp/comfyui_user \
+    /tmp/comfyui_input \
+    /tmp/comfyui_output \
+    /tmp/comfyui_temp \
+    /tmp/workflows \
+    /tmp/comfyui_models \
+    || echo "Failed to set permissions, continuing anyway"
 
 # Copy all required application files
 COPY requirements.txt .
@@ -54,11 +69,12 @@ COPY app.py .
 COPY start.sh .
 COPY download_models.py .
 
-# Try to copy workflow files - if these fail, they'll be downloaded by start.sh
+# Copy workflows if they exist, otherwise they'll be downloaded in start.sh
+# Use a proper directory path with a trailing slash
 COPY workflows/ /app/workflows/
 
-# Instead of trying to copy LFS files in the Dockerfile, we'll let start.sh handle downloading them
-# This ensures we always get the complete files rather than LFS pointers
+# Initialize the PuLID directory - the actual content will be downloaded from Hugging Face
+RUN echo "PuLID nodes will be downloaded at runtime" > /app/ComfyUI/custom_nodes/PuLID/README.txt
 
 # Make scripts executable
 RUN chmod +x /app/start.sh
@@ -66,8 +82,8 @@ RUN chmod +x /app/start.sh
 # Install Python dependencies
 RUN pip3 install --no-cache-dir -r requirements.txt
 
-# Run model download script
-RUN python3 download_models.py || true
+# Run model download script (will be overridden at runtime)
+RUN python3 download_models.py || echo "Initial model download failed, will retry at runtime"
 
 # Expose ports for Gradio and ComfyUI
 EXPOSE 7860 8188