egutierrez
cda36408d0
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>
81 lines
3.0 KiB
Python
81 lines
3.0 KiB
Python
"""Envia un workflow (API format) a un servidor ComfyUI via POST /prompt.
|
|
|
|
Funcion impura: hace red (HTTP POST). Solo stdlib (urllib, json, uuid).
|
|
|
|
ComfyUI encola el workflow y devuelve un dict con prompt_id (para seguir el
|
|
resultado con comfyui_wait_result), number (posicion en la cola) y node_errors.
|
|
Si el workflow es invalido, ComfyUI responde HTTP 400 con detalles de la
|
|
validacion por nodo: esta funcion los captura y los propaga en el error.
|
|
"""
|
|
import json
|
|
import urllib.error
|
|
import urllib.request
|
|
import uuid
|
|
|
|
|
|
def comfyui_submit_workflow(
|
|
workflow: dict,
|
|
server: str = "127.0.0.1:8188",
|
|
client_id: str | None = None,
|
|
timeout: float = 30.0,
|
|
) -> dict:
|
|
"""Encola un workflow en ComfyUI y devuelve la respuesta del servidor.
|
|
|
|
Args:
|
|
workflow: dict en API format (resultado de
|
|
comfyui_build_txt2img_workflow u otro builder). Las claves son
|
|
node_ids y cada valor tiene class_type + inputs.
|
|
server: host:port del servidor ComfyUI (sin esquema).
|
|
client_id: identificador de cliente para correlar eventos (WS/history).
|
|
Si es None, se genera un uuid4. Se incluye en la respuesta.
|
|
timeout: timeout de la peticion HTTP en segundos.
|
|
|
|
Returns:
|
|
dict con al menos prompt_id (str), number (int, posicion en cola) y
|
|
node_errors (dict). Se anade la clave "client_id" usada.
|
|
|
|
Raises:
|
|
RuntimeError: si ComfyUI rechaza el workflow (HTTP 400 con detalles de
|
|
validacion en el cuerpo), si no se puede conectar, o si la respuesta
|
|
no es JSON valido. El mensaje incluye el cuerpo del error.
|
|
"""
|
|
cid = client_id or str(uuid.uuid4())
|
|
body = json.dumps({"prompt": workflow, "client_id": cid}).encode()
|
|
url = f"http://{server}/prompt"
|
|
req = urllib.request.Request(
|
|
url, data=body, headers={"Content-Type": "application/json"}
|
|
)
|
|
try:
|
|
with urllib.request.urlopen(req, timeout=timeout) as resp:
|
|
out = json.loads(resp.read())
|
|
except urllib.error.HTTPError as exc:
|
|
detail = exc.read().decode(errors="replace")
|
|
raise RuntimeError(
|
|
f"comfyui_submit_workflow: ComfyUI rechazo el workflow "
|
|
f"(HTTP {exc.code}): {detail}"
|
|
) from exc
|
|
except urllib.error.URLError as exc:
|
|
raise RuntimeError(
|
|
f"comfyui_submit_workflow: no se pudo conectar a {url}: {exc.reason}"
|
|
) from exc
|
|
except json.JSONDecodeError as exc:
|
|
raise RuntimeError(
|
|
f"comfyui_submit_workflow: respuesta no es JSON valido desde {url}: {exc}"
|
|
) from exc
|
|
out["client_id"] = cid
|
|
return out
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from comfyui_build_txt2img_workflow import comfyui_build_txt2img_workflow
|
|
|
|
wf = comfyui_build_txt2img_workflow(
|
|
ckpt_name="IMG_v1-5-pruned-emaonly-fp16.safetensors",
|
|
positive="a red apple on a wooden table, sharp focus",
|
|
negative="blurry, low quality",
|
|
steps=20,
|
|
seed=42,
|
|
)
|
|
resp = comfyui_submit_workflow(wf)
|
|
print(f"prompt_id={resp['prompt_id']} number={resp.get('number')}")
|