fix: auto-detect RunPod environment and use correct paths
- Automatically detects RunPod environment (checks for /workspace directory) - Uses RunPod paths by default: /workspace/huggingface_cache and /workspace/ComfyUI/models - Falls back to local paths (~/.cache/huggingface and ~/ComfyUI/models) on non-RunPod systems - Updated help text to show both RunPod and local defaults - Updated example YAML with RunPod paths 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -97,11 +97,16 @@ else
|
|||||||
CONFIG_FILE="" # No config file by default
|
CONFIG_FILE="" # No config file by default
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Default cache directory (use HuggingFace default)
|
# Default cache directory - detect RunPod or use HuggingFace default
|
||||||
|
if [[ -d "/workspace" ]]; then
|
||||||
|
# RunPod environment
|
||||||
|
CACHE_DIR="${CACHE_DIR:-/workspace/huggingface_cache}"
|
||||||
|
COMFYUI_DIR="${COMFYUI_DIR:-/workspace/ComfyUI/models}"
|
||||||
|
else
|
||||||
|
# Local environment
|
||||||
CACHE_DIR="${CACHE_DIR:-${HOME}/.cache/huggingface}"
|
CACHE_DIR="${CACHE_DIR:-${HOME}/.cache/huggingface}"
|
||||||
|
|
||||||
# Default ComfyUI models directory
|
|
||||||
COMFYUI_DIR="${COMFYUI_DIR:-${HOME}/ComfyUI/models}"
|
COMFYUI_DIR="${COMFYUI_DIR:-${HOME}/ComfyUI/models}"
|
||||||
|
fi
|
||||||
|
|
||||||
# Default command
|
# Default command
|
||||||
COMMAND="both"
|
COMMAND="both"
|
||||||
@@ -727,8 +732,12 @@ while [[ $# -gt 0 ]]; do
|
|||||||
echo ""
|
echo ""
|
||||||
echo "Options:"
|
echo "Options:"
|
||||||
echo " -c, --config FILE Configuration file (default: NONE)"
|
echo " -c, --config FILE Configuration file (default: NONE)"
|
||||||
echo " --cache-dir DIR Cache directory (default: ~/.cache/huggingface)"
|
echo " --cache-dir DIR Cache directory (default: auto-detect)"
|
||||||
echo " --comfyui-dir DIR ComfyUI models directory (default: ~/ComfyUI/models)"
|
echo " RunPod: /workspace/huggingface_cache"
|
||||||
|
echo " Local: ~/.cache/huggingface"
|
||||||
|
echo " --comfyui-dir DIR ComfyUI models directory (default: auto-detect)"
|
||||||
|
echo " RunPod: /workspace/ComfyUI/models"
|
||||||
|
echo " Local: ~/ComfyUI/models"
|
||||||
echo " -h, --help Show this help message"
|
echo " -h, --help Show this help message"
|
||||||
echo ""
|
echo ""
|
||||||
echo "Examples:"
|
echo "Examples:"
|
||||||
|
|||||||
112
comfyui_models.example.yaml
Normal file
112
comfyui_models.example.yaml
Normal file
@@ -0,0 +1,112 @@
|
|||||||
|
# ComfyUI Models Configuration Example
|
||||||
|
#
|
||||||
|
# This file defines which models to download from HuggingFace
|
||||||
|
# and explicit source->destination file mappings for symlinks.
|
||||||
|
#
|
||||||
|
# IMPORTANT: Use the 'files' array to specify EXACTLY which files to link
|
||||||
|
# and where to link them. Without explicit file mappings, ALL files from
|
||||||
|
# the repository will be linked (which causes trash/wrong files).
|
||||||
|
#
|
||||||
|
# Format:
|
||||||
|
# - repo_id: HuggingFace repository (e.g., "runwayml/stable-diffusion-v1-5")
|
||||||
|
# - description: Human-readable description
|
||||||
|
# - size_gb: Approximate size
|
||||||
|
# - essential: Is this a core model?
|
||||||
|
# - type: ComfyUI subdirectory (checkpoints, vae, loras, controlnet, etc.)
|
||||||
|
# - filename: Optional filter for find_model_files (usually leave empty)
|
||||||
|
# - files: Array of {source: "filename_in_repo", dest: "filename_in_comfyui"}
|
||||||
|
#
|
||||||
|
# Example:
|
||||||
|
# files:
|
||||||
|
# - source: "v1-5-pruned-emaonly.safetensors"
|
||||||
|
# dest: "sd-v1-5.safetensors"
|
||||||
|
#
|
||||||
|
# This will create: ComfyUI/models/checkpoints/sd-v1-5.safetensors
|
||||||
|
# Pointing to: ~/.cache/huggingface/.../v1-5-pruned-emaonly.safetensors
|
||||||
|
|
||||||
|
settings:
|
||||||
|
# RunPod: /workspace/huggingface_cache
|
||||||
|
# Local: ~/.cache/huggingface
|
||||||
|
cache_dir: /workspace/huggingface_cache
|
||||||
|
parallel_downloads: 1
|
||||||
|
|
||||||
|
model_categories:
|
||||||
|
# Stable Diffusion Checkpoints
|
||||||
|
checkpoints:
|
||||||
|
- repo_id: runwayml/stable-diffusion-v1-5
|
||||||
|
description: Stable Diffusion v1.5
|
||||||
|
size_gb: 4
|
||||||
|
essential: true
|
||||||
|
type: checkpoints
|
||||||
|
filename: ""
|
||||||
|
files:
|
||||||
|
- source: "v1-5-pruned-emaonly.safetensors"
|
||||||
|
dest: "sd-v1-5.safetensors"
|
||||||
|
|
||||||
|
- repo_id: stabilityai/stable-diffusion-xl-base-1.0
|
||||||
|
description: SDXL Base 1.0
|
||||||
|
size_gb: 7
|
||||||
|
essential: true
|
||||||
|
type: checkpoints
|
||||||
|
filename: ""
|
||||||
|
files:
|
||||||
|
- source: "sd_xl_base_1.0.safetensors"
|
||||||
|
dest: "sdxl-base-1.0.safetensors"
|
||||||
|
|
||||||
|
# VAE Models
|
||||||
|
vae:
|
||||||
|
- repo_id: stabilityai/sd-vae-ft-mse-original
|
||||||
|
description: SD VAE ft MSE
|
||||||
|
size_gb: 0.3
|
||||||
|
essential: true
|
||||||
|
type: vae
|
||||||
|
filename: ""
|
||||||
|
files:
|
||||||
|
- source: "vae-ft-mse-840000-ema-pruned.safetensors"
|
||||||
|
dest: "sd-vae-ft-mse.safetensors"
|
||||||
|
|
||||||
|
# LoRA Models
|
||||||
|
loras:
|
||||||
|
- repo_id: latent-consistency/lcm-lora-sdv1-5
|
||||||
|
description: LCM LoRA for SD v1.5
|
||||||
|
size_gb: 0.1
|
||||||
|
essential: false
|
||||||
|
type: loras
|
||||||
|
filename: ""
|
||||||
|
files:
|
||||||
|
- source: "pytorch_lora_weights.safetensors"
|
||||||
|
dest: "lcm-lora-sdv1-5.safetensors"
|
||||||
|
|
||||||
|
# ControlNet Models
|
||||||
|
controlnet:
|
||||||
|
- repo_id: lllyasviel/control_v11p_sd15_canny
|
||||||
|
description: ControlNet Canny
|
||||||
|
size_gb: 1.4
|
||||||
|
essential: false
|
||||||
|
type: controlnet
|
||||||
|
filename: ""
|
||||||
|
files:
|
||||||
|
- source: "diffusion_pytorch_model.safetensors"
|
||||||
|
dest: "control_v11p_sd15_canny.safetensors"
|
||||||
|
|
||||||
|
- repo_id: lllyasviel/control_v11p_sd15_openpose
|
||||||
|
description: ControlNet OpenPose
|
||||||
|
size_gb: 1.4
|
||||||
|
essential: false
|
||||||
|
type: controlnet
|
||||||
|
filename: ""
|
||||||
|
files:
|
||||||
|
- source: "diffusion_pytorch_model.safetensors"
|
||||||
|
dest: "control_v11p_sd15_openpose.safetensors"
|
||||||
|
|
||||||
|
# Upscale Models
|
||||||
|
upscale_models:
|
||||||
|
- repo_id: ai-forever/Real-ESRGAN
|
||||||
|
description: Real-ESRGAN x4
|
||||||
|
size_gb: 0.1
|
||||||
|
essential: false
|
||||||
|
type: upscale_models
|
||||||
|
filename: ""
|
||||||
|
files:
|
||||||
|
- source: "RealESRGAN_x4plus.pth"
|
||||||
|
dest: "RealESRGAN_x4plus.pth"
|
||||||
Reference in New Issue
Block a user