Upload start.sh with huggingface_hub

start.sh

@@ -14,10 +14,10 @@ mkdir -p /tmp/comfyui_output
 mkdir -p /tmp/comfyui_temp
 mkdir -p /tmp/comfyui_user
 mkdir -p /tmp/workflows
-mkdir -p /tmp/comfyui_models
+mkdir -p /tmp/comfyui_models/pulid
 
-# Set proper permissions
-chmod -R 777 /tmp/comfyui_input /tmp/comfyui_output /tmp/comfyui_temp /tmp/comfyui_user /tmp/workflows /tmp/comfyui_models
+# Try to set permissions, but don't fail if we can't
+chmod -R 777 /tmp/comfyui_input /tmp/comfyui_output /tmp/comfyui_temp /tmp/comfyui_user /tmp/workflows /tmp/comfyui_models || echo "Failed to set permissions, continuing anyway"
 
 # Download files from the dataset repository
 echo "Downloading images from dataset repository..."
@@ -32,19 +32,77 @@ echo "Verifying downloaded files:"
 ls -la /tmp/comfyui_input/
 ls -la /tmp/workflows/
 
-# Run model downloader
-echo "Downloading models..."
+# Run model downloader to get models and PuLID
+echo "Downloading models and PuLID code..."
 cd /app
 python3 download_models.py
 
-# Set environment variables
+# Setup folder paths for ComfyUI
+cat > /tmp/folder_paths.json << EOF
+{
+    "input_path": "/tmp/comfyui_input",
+    "output_path": "/tmp/comfyui_output",
+    "temp_path": "/tmp/comfyui_temp",
+    "models_path": {
+        "checkpoints": ["/tmp/comfyui_models/checkpoints"],
+        "controlnet": ["/tmp/comfyui_models/controlnet"],
+        "clip_vision": ["/tmp/comfyui_models/clip_vision"],
+        "ipadapter": ["/tmp/comfyui_models/ipadapter"],
+        "pulid": ["/tmp/comfyui_models/pulid"]
+    },
+    "user_path": "/tmp/comfyui_user"
+}
+EOF
+
+# Set environment variables to ensure we find all modules
 export PYTHONPATH="${PYTHONPATH}:/app/ComfyUI:/app/ComfyUI/custom_nodes"
+export COMFYUI_FOLDER_PATHS_FILE="/tmp/folder_paths.json"
 
-# Install additional Python packages
+# Install additional Python packages (but don't fail if they can't be installed)
 echo "Installing required Python packages..."
-pip install -e /app/ComfyUI/custom_nodes/PuLID || echo "Failed to install PuLID, continuing anyway"
 pip install comfyui-frontend-package comfyui-workflow-templates || echo "Package installation failed, continuing anyway"
 
+# Check if PuLID is properly set up
+if [ -f "/app/ComfyUI/custom_nodes/PuLID/__init__.py" ]; then
+    echo "PuLID node found at /app/ComfyUI/custom_nodes/PuLID"
+else
+    echo "WARNING: PuLID node not found! Creating minimal implementation..."
+    mkdir -p /app/ComfyUI/custom_nodes/PuLID
+    
+    # Create minimal init file
+    cat > /app/ComfyUI/custom_nodes/PuLID/__init__.py << EOF
+from .pulid_node import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+EOF
+
+    # Create minimal node implementation
+    cat > /app/ComfyUI/custom_nodes/PuLID/pulid_node.py << EOF
+import os
+import folder_paths
+
+class PulidModelLoader:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": {"model_name": (folder_paths.get_filename_list("pulid"), )}}
+    
+    RETURN_TYPES = ("PULID_MODEL",)
+    FUNCTION = "load_model"
+    CATEGORY = "loaders"
+
+    def load_model(self, model_name):
+        model_path = folder_paths.get_full_path("pulid", model_name)
+        return (model_path,)
+
+NODE_CLASS_MAPPINGS = {
+    "PulidModelLoader": PulidModelLoader
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "PulidModelLoader": "Load PuLID Model"
+}
+EOF
+    echo "Created minimal PuLID implementation"
+fi
+
 # Start ComfyUI in the background with custom directories
 cd /app/ComfyUI
 echo "Starting ComfyUI server..."
@@ -52,8 +110,7 @@ python3 main.py --listen 0.0.0.0 --port 8188 \
   --input-directory /tmp/comfyui_input \
   --output-directory /tmp/comfyui_output \
   --temp-directory /tmp/comfyui_temp \
-  --user-dir /tmp/comfyui_user \
-  --models-directory /tmp/comfyui_models &
+  --user-dir /tmp/comfyui_user &
 COMFY_PID=$!
 
 # Wait for ComfyUI to start