Acly
ffa130239b
* from kohya-ss/ComfyUI-Anima-LLLite * split nodes into load/apply to cache model loading
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from comfy_api.latest import ComfyExtension, io
|
|
|
|
from . import api as api
|
|
from . import control, krita, nodes, region, tile, translation
|
|
|
|
|
|
class ExternalToolingNodes(ComfyExtension):
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
node_list = [
|
|
nodes.LoadImageCache,
|
|
nodes.SaveImageCache,
|
|
nodes.LoadImageBase64,
|
|
nodes.LoadMaskBase64,
|
|
nodes.SendImageWebSocket,
|
|
nodes.ApplyMaskToImage,
|
|
nodes.ReferenceImage,
|
|
nodes.ApplyReferenceImages,
|
|
tile.CreateTileLayout,
|
|
tile.ExtractImageTile,
|
|
tile.ExtractMaskTile,
|
|
tile.GenerateTileMask,
|
|
tile.MergeImageTile,
|
|
region.BackgroundRegion,
|
|
region.DefineRegion,
|
|
region.ListRegionMasks,
|
|
region.AttentionMask,
|
|
translation.Translate,
|
|
krita.KritaOutput,
|
|
krita.KritaSendText,
|
|
krita.KritaCanvas,
|
|
krita.KritaSelection,
|
|
krita.KritaImageLayer,
|
|
krita.KritaMaskLayer,
|
|
krita.Parameter,
|
|
krita.KritaStyle,
|
|
krita.KritaStyleAndPrompt,
|
|
control.ControlApply,
|
|
control.ControlLoad,
|
|
]
|
|
try: # see #66
|
|
from . import nsfw
|
|
|
|
node_list.append(nsfw.NSFWFilter)
|
|
except (ImportError, ModuleNotFoundError):
|
|
import traceback
|
|
|
|
print("[comfyui-tooling-nodes] WARNING: Could not import all nodes.")
|
|
traceback.print_exc()
|
|
|
|
return node_list
|
|
|
|
|
|
async def comfy_entrypoint():
|
|
return ExternalToolingNodes()
|
|
|
|
|
|
WEB_DIRECTORY = "./js"
|