70d541fca9
Grupo nuevo comfyui-skill: recetas versionadas de generación ComfyUI que
compilan a un workflow cambiando solo el subject.
- comfyui_build_skill_workflow (pura): receta -> workflow API format,
despacha base (txt2img/flux/sdxl_refiner), sustituye {subject}+triggers,
encadena loras e inject blocks (facedetailer, hires_fix). SkillWorkflowError tipada.
- comfyui_inject_hires_fix (pura): inyecta 2ª pasada UltimateSDUpscale sobre dict.
- comfyui_save/load/list_skill (impuras): CRUD de la librería en disco con
versionado por snapshots, round-trip idéntico, filtro NSFW.
- ask_llm_vision (core, claude-direct): pregunta multimodal imagen+texto via
API directa Anthropic, para puntuar generaciones.
- Página madre docs/capabilities/comfyui-skill.md con schema canónico de recipe.json.
Tests offline: 11 verdes (6 builder + 5 inject_hires_fix). Sin GPU.
4.9 KiB
4.9 KiB
name, kind, lang, domain, version, purity, signature, description, tags, uses_functions, uses_types, returns, returns_optional, error_type, imports, params, output, tested, tests, test_file_path, file_path
| name | kind | lang | domain | version | purity | signature | description | tags | uses_functions | uses_types | returns | returns_optional | error_type | imports | params | output | tested | tests | test_file_path | file_path | |||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| comfyui_inject_hires_fix | function | py | ml | 1.0.0 | pure | def comfyui_inject_hires_fix(workflow: dict, *, upscale_by: float = 1.5, denoise: float = 0.4, steps: int = 20, cfg: float = 7.0, seed: int = 0, upscale_model: str = '4x_foolhardy_Remacri.pth', sampler_name: str = 'euler', scheduler: str = 'normal', tile_width: int = 512, tile_height: int = 512) -> dict | Inyecta una segunda pasada hires-fix en un workflow ComfyUI ya construido (API format) que termina en VAEDecode -> SaveImage. Anade UpscaleModelLoader + UltimateSDUpscale (re-difusion por tiles) conectados a la imagen del VAEDecode y al model/vae del CheckpointLoaderSimple, y repunta el SaveImage a la imagen ampliada. Version encadenable-sobre-dict de comfyui_build_hires_fix_workflow. Pura: no muta el dict de entrada (copia profunda). |
|
false |
|
copia del workflow con UpscaleModelLoader + UltimateSDUpscale anadidos (node_ids = max id numerico + 1 y + 2) y el SaveImage repuntado a la salida [ultimatesdupscale_id, 0]. Si no habia SaveImage, se anade uno con filename_prefix 'hires'. | true |
|
python/functions/ml/tests/test_comfyui_inject_hires_fix.py | python/functions/ml/comfyui_inject_hires_fix.py |
Ejemplo
import sys, os
sys.path.insert(0, os.path.join(os.environ["HOME"], "fn_registry", "python", "functions"))
from ml.comfyui_build_txt2img_workflow import comfyui_build_txt2img_workflow
from ml.comfyui_inject_hires_fix import comfyui_inject_hires_fix
base = comfyui_build_txt2img_workflow("dreamshaper_8.safetensors", "a fox in a forest, detailed")
# El base termina en VAEDecode ["8",0] -> SaveImage ["9"].
wf = comfyui_inject_hires_fix(base, upscale_by=2.0, denoise=0.35, seed=42)
# Ahora: VAEDecode -> UltimateSDUpscale -> SaveImage, con UpscaleModelLoader cargando Remacri.
# El SaveImage["9"].inputs["images"] apunta al nuevo UltimateSDUpscale, no al VAEDecode.
Cuando usarla
Cuando ya tengas un workflow txt2img/img2img construido (o devuelto por otro
builder) y quieras anadirle el hires fix sin reescribir el grafo desde cero.
A diferencia de comfyui_build_hires_fix_workflow, que construye el grafo entero
de una vez, esta lo ENCADENA sobre un dict existente: util tras inyectar LoRAs
con comfyui_inject_lora o partiendo de cualquier base que termine en
VAEDecode -> SaveImage. Una sola llamada anade la segunda pasada completa.
Gotchas
- Pura: no muta el
workflowde entrada (trabaja sobre una copia profunda) y NO valida queupscale_modelexista en el servidor. Valida concomfyui_validate_workflow. - Requiere el custom node UltimateSDUpscale instalado en el servidor ComfyUI; el dict se construye igual aunque no este, pero el submit fallara.
- Detecta el VAEDecode (fuente de imagen), el CheckpointLoaderSimple (model slot 0, vae slot 2) y los CLIPTextEncode positive/negative por el KSampler existente. Si no hay VAEDecode o CheckpointLoaderSimple, lanza ValueError.
- Si el workflow tiene varios VAEDecode/SaveImage, se usa el PRIMERO encontrado. Para grafos multi-salida construye con un builder dedicado.
- El nuevo node_id es
max(ids numericos) + 1(y +2). Si tu workflow usa ids no numericos, el contador cae alen(workflow) + 1.