diff --git a/arty.yml b/arty.yml index 6ed6fd1..ae5c63e 100644 --- a/arty.yml +++ b/arty.yml @@ -518,20 +518,57 @@ scripts: ln -sf /workspace/huggingface_cache/models--stabilityai--stable-video-diffusion-img2vid-xt stable-video-diffusion-img2vid-xt echo "Models linked to ComfyUI" - # Workflow linking (link production workflows from git repo) + # Workflow linking (link production workflows from git repo - flattened structure) workflows/link-comfyui: | mkdir -p /workspace/ComfyUI/workflows cd /workspace/ComfyUI/workflows - ln -sf /workspace/ai/models/comfyui/workflows/text-to-image text-to-image - ln -sf /workspace/ai/models/comfyui/workflows/image-to-image image-to-image - ln -sf /workspace/ai/models/comfyui/workflows/image-to-video image-to-video - ln -sf /workspace/ai/models/comfyui/workflows/text-to-music text-to-music - ln -sf /workspace/ai/models/comfyui/workflows/upscaling upscaling - ln -sf /workspace/ai/models/comfyui/workflows/advanced advanced - ln -sf /workspace/ai/models/comfyui/workflows/templates templates - ln -sf /workspace/ai/models/comfyui/workflows/README.md README.md - ln -sf /workspace/ai/models/comfyui/workflows/WORKFLOW_STANDARDS.md WORKFLOW_STANDARDS.md - echo "Production workflows linked to ComfyUI" + + # Remove old directory symlinks if they exist + rm -f text-to-image image-to-image image-to-video text-to-music upscaling advanced templates + + # Link all JSON workflow files directly to workflows root + SOURCE_DIR="/workspace/ai/models/comfyui/workflows" + + # Text-to-Image workflows + for file in "$SOURCE_DIR/text-to-image"/*.json; do + [ -f "$file" ] && ln -sf "$file" "$(basename "$file")" + done + + # Image-to-Image workflows + for file in "$SOURCE_DIR/image-to-image"/*.json; do + [ -f "$file" ] && ln -sf "$file" "$(basename "$file")" + done + + # Image-to-Video workflows + for file in "$SOURCE_DIR/image-to-video"/*.json; do + [ -f "$file" ] && ln -sf "$file" "$(basename "$file")" + done + + # Text-to-Music workflows + for file in "$SOURCE_DIR/text-to-music"/*.json; do + [ -f "$file" ] && ln -sf "$file" "$(basename "$file")" + done + + # Upscaling workflows + for file in "$SOURCE_DIR/upscaling"/*.json; do + [ -f "$file" ] && ln -sf "$file" "$(basename "$file")" + done + + # Advanced workflows + for file in "$SOURCE_DIR/advanced"/*.json; do + [ -f "$file" ] && ln -sf "$file" "$(basename "$file")" + done + + # Link documentation files + ln -sf "$SOURCE_DIR/README.md" README.md + ln -sf "$SOURCE_DIR/WORKFLOW_STANDARDS.md" WORKFLOW_STANDARDS.md + + # Count workflows + WORKFLOW_COUNT=$(ls -1 *.json 2>/dev/null | wc -l) + + echo "Production workflows linked to ComfyUI (flattened structure)" + echo " Total workflows: $WORKFLOW_COUNT" + echo "" echo " - 4 text-to-image workflows (FLUX, SDXL, SD3.5)" echo " - 3 image-to-image workflows (IP-Adapter)" echo " - 3 image-to-video workflows (CogVideoX, SVD)"