27 lines
922 B
Python
27 lines
922 B
Python
"""ComfyUI extension entrypoint for ComfyUI-JoyAI.
|
|
|
|
Installs the JoyAI-Image-Edit integration hooks at import time (before Comfy
|
|
scans for checkpoints) so a JoyAI-Image-Edit transformer in
|
|
``models/diffusion_models/`` and the Qwen3-VL-8B encoder in
|
|
``models/text_encoders/`` are recognized by the stock ``UNETLoader`` /
|
|
``CLIPLoader``. Exposes the JoyImage edit conditioning node.
|
|
"""
|
|
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
from .joyimage.integration import register as register_joyimage
|
|
from .joyimage.nodes import NODE_CLASSES
|
|
|
|
# Idempotent: hooks must be live before Comfy's startup checkpoint scan so the
|
|
# transformer shows up in the UNETLoader dropdown on first launch.
|
|
register_joyimage()
|
|
|
|
|
|
class JoyAIExtension(ComfyExtension):
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return list(NODE_CLASSES)
|
|
|
|
|
|
async def comfy_entrypoint() -> ComfyExtension:
|
|
return JoyAIExtension()
|