Files
fn_registry/python/functions/ml/comfyui_inject_hires_fix.md
T
egutierrez cda36408d0 feat(ml): modelos con prefijo de categoría (IMG_/VIDEO_/3D_) + refs actualizadas
Renombra los 13 checkpoints/diffusion models de ComfyUI prefijando la
categoría al inicio del nombre, para que en el dropdown de carga el usuario
distinga de inmediato imagen/vídeo/3D y no cargue un modelo en el nodo
equivocado. Misma operación que se hizo con los LoRAs (report 0197) pero
sobre los modelos.

Clasificación:
- IMG_: dreamshaper_8, juggernaut_xl_v11, v1-5-pruned-emaonly-fp16,
  flux1-dev-fp8-e4m3fn, flux1-schnell-fp8-e4m3fn
- VIDEO_: svd, ltx-video-2b-v0.9.5, wan2.1_t2v_1.3B_fp16
- 3D_: stable_zero123, sv3d_p, hunyuan3d-dit-v2-mini, hunyuan3d-dit-v2-mv,
  hy3dgen/hunyuan3d-dit-v2-0-fp16 (mantiene subcarpeta)

A diferencia de los LoRAs aquí solo se PREFIJA la categoría conservando el
nombre completo (versión/arquitectura). Archivos físicos renombrados en
~/ComfyUI/models/checkpoints, /mnt/2tb/comfyui_models/{checkpoints,
diffusion_models} y la subcarpeta hy3dgen/. Mapa de reversión en
~/ComfyUI/models/checkpoints/_ckpt_rename_map.json.

Actualiza todas las refs (ckpt_name/unet_name + defaults + prosa) en los
builders gamedev/vídeo/3D, style presets, pipelines, tests y los workflows
de ComfyUI. Arregla de paso el default roto de comfyui_text_to_3d_oneshot
(apuntaba a v1-5-pruned-emaonly.safetensors inexistente; ahora al real
IMG_v1-5-pruned-emaonly-fp16.safetensors).

No tocados (justificado): repo-paths de HuggingFace en comfyui_install_3d_model
(<repo>/model.fp16.safetensors son rutas de descarga, no nombres de dropdown)
y el mock de stable-diffusion.cpp en test_genconfig_to_sdcpp_args.

Verificado: dropdowns CheckpointLoaderSimple + UNETLoader listan los nombres
con prefijo; 1 generación real con IMG_juggernaut_xl_v11 (node_errors vacío,
pixelart_00003_.png); 327 tests comfyui verdes.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
2026-06-27 18:24:52 +02:00

5.4 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.1 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).
comfyui
comfyui-skill
ml
hires-fix
upscale
workflow
stable-diffusion
false
name desc
workflow dict en API format (ej. salida de comfyui_build_txt2img_workflow) que termina en VAEDecode -> SaveImage. No se muta; se devuelve una copia.
name desc
upscale_by Factor de ampliacion de UltimateSDUpscale sobre la imagen base (1.5 -> 512 pasa a 768). keyword-only.
name desc
denoise Fuerza de re-difusion de la segunda pasada (0.4 por defecto). <1 conserva la composicion base y solo anade detalle; 1.0 la re-generaria entera. keyword-only.
name desc
steps Pasos de sampling de la re-difusion tiled. keyword-only.
name desc
cfg Classifier-free guidance de la re-difusion. keyword-only.
name desc
seed Semilla de UltimateSDUpscale. keyword-only.
name desc
upscale_model Modelo de upscale en models/upscale_models/ que usa UltimateSDUpscale para escalar antes de re-difundir (ej. '4x_foolhardy_Remacri.pth'). keyword-only.
name desc
sampler_name Sampler de la re-difusion. keyword-only.
name desc
scheduler Scheduler de la re-difusion. keyword-only.
name desc
tile_width Ancho de tile de UltimateSDUpscale (px). Tiles mas pequenos = menos VRAM, mas costuras. keyword-only.
name desc
tile_height Alto de tile de UltimateSDUpscale (px). keyword-only.
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
no muta el dict de entrada (pureza)
inserta UltimateSDUpscale y UpscaleModelLoader
repunta el SaveImage al UltimateSDUpscale
params reflejados (upscale_by/denoise/seed) + batch_size
lanza ValueError si falta VAEDecode
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("IMG_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 workflow de entrada (trabaja sobre una copia profunda) y NO valida que upscale_model exista en el servidor. Valida con comfyui_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 a len(workflow) + 1.
  • El nodo UltimateSDUpscale declara batch_size como input requerido en /object_info; la inyección lo fija a 1. Sin él, el submit pasaba la validación de POST pero el grafo fallaba en runtime por input faltante.

Capability growth log

  • v1.0.1 (2026-06-27) — bugfix paralelo al de comfyui_build_hires_fix_workflow: el nodo UltimateSDUpscale exige batch_size (input requerido en /object_info); se añade batch_size: 1 al nodo inyectado para que el workflow no falle en runtime.