python ~/ComfyUI/_install.models.etc.py [path to workflow.json] <br> installs all missing models and lets you insert civitai key if needed <br> extract workflow from image using https://absurdity.ai/comfyui/image2workflowjson.html

readme.md

python ~/ComfyUI/_install.models.etc.py [path to workflow.json]

installs all missing models and lets you insert civitai or huggingface key if needed

extract workflow json from image using https://absurdity.ai/comfyui/image2workflowjson.html

install.models.etc.py

import json
import os
import sys
import requests

MODELS_BASE_DIR = os.path.join("ComfyUI", "models")

def download_file(url, directory, filename, civitai_key=None, hf_token=None):
    target_dir = os.path.join(MODELS_BASE_DIR, directory)
    os.makedirs(target_dir, exist_ok=True)
    target_path = os.path.join(target_dir, filename)

    if os.path.exists(target_path):
        print(f"✅ Already exists: {target_path}")
        return

    print(f"⬇️ Downloading {filename} → {directory}")

    headers = {}
    if civitai_key and "civitai.com" in url:
        headers["Authorization"] = f"Bearer {civitai_key}"
    if hf_token and "huggingface.co" in url:
        headers["Authorization"] = f"Bearer {hf_token}"

    try:
        with requests.get(url, stream=True, headers=headers) as r:
            r.raise_for_status()
            with open(target_path, "wb") as f:
                for chunk in r.iter_content(chunk_size=8192):
                    if chunk:
                        f.write(chunk)
        print(f"✅ Saved: {target_path}")
    except Exception as e:
        print(f"❌ Failed to download {url}: {e}")

def download_models_from_list(models, civitai_key=None, hf_token=None):
    for m in models:
        name = m.get("name")
        url = m.get("url")
        directory = m.get("directory", "")
        if name and url and directory:
            download_file(url, directory, name, civitai_key, hf_token)
        else:
            print(f"⚠️ Missing info for model: {m}")

def main(workflow_path):
    if not os.path.isfile(workflow_path):
        print(f"❌ File not found: {workflow_path}")
        sys.exit(1)

    with open(workflow_path, "r", encoding="utf-8") as f:
        workflow_data = json.load(f)

    # Collect all model URLs from root models and node properties
    all_urls = set()
    for m in workflow_data.get("models", []):
        url = m.get("url", "")
        if url:
            all_urls.add(url)
    for node in workflow_data.get("nodes", []):
        props = node.get("properties", {})
        models = props.get("models", [])
        for m in models:
            url = m.get("url", "")
            if url:
                all_urls.add(url)

    civitai_key = None
    if any("civitai.com" in url for url in all_urls):
        civitai_key = input("🔑 Enter your Civitai API key (or leave empty to skip): ").strip() or None

    hf_token = None
    if any("huggingface.co" in url for url in all_urls):
        hf_token = input("🔑 Enter your Huggingface API token (or leave empty to skip): ").strip() or None

    print("⏳ Downloading models listed at root...")
    download_models_from_list(workflow_data.get("models", []), civitai_key, hf_token)

    print("⏳ Downloading models listed inside nodes...")
    for node in workflow_data.get("nodes", []):
        props = node.get("properties", {})
        models = props.get("models", [])
        download_models_from_list(models, civitai_key, hf_token)

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: python download_comfyui_models.py <workflow.json>")
        sys.exit(1)

    main(sys.argv[1])