mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-20 22:17:31 +08:00
d71cc1c8f2
* Move detection category under image category * Add missing categories * Move detection nodes to detection category * Move save nodes to image root catefory * Rename postprocessors * Move mask category under image * Move guiders category to parent level at root of sampling category * Move custom_sampling category to parent level at the root of sampling category * Modify description of LoRA loaders * Fix node id SolidMask * Move VOID Quadmask under image/mask * Group compositing nodes under image/compositing * Move load image as mask to image category for consistency with other load image nodes * Align display name with Load Checkpoint * Move dataset category under training category * Rename Number Convert to Conver Number (verb first) * Rename Canny node * Revert wanBlockSwap + description * Add description to RemoveBackground node * Revert category update of dataset
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from comfy_api.latest import ComfyExtension, io
|
|
from typing_extensions import override
|
|
# If you write a node that is so useless that it breaks ComfyUI it will be featured in this exclusive list
|
|
|
|
# "native" block swap nodes are placebo at best and break the ComfyUI memory management system.
|
|
# They are also considered harmful because instead of users reporting issues with the built in
|
|
# memory management they install these stupid nodes and complain even harder. Now it completely
|
|
# breaks with some of the new ComfyUI memory optimizations so I have made the decision to NOP it
|
|
# out of all workflows.
|
|
class wanBlockSwap(io.ComfyNode):
|
|
@classmethod
|
|
def define_schema(cls):
|
|
return io.Schema(
|
|
node_id="wanBlockSwap",
|
|
category="",
|
|
description="Intercept wanBlockSwap custom node that causes major instability and make it no-op.",
|
|
inputs=[
|
|
io.Model.Input("model"),
|
|
],
|
|
outputs=[
|
|
io.Model.Output(),
|
|
],
|
|
is_deprecated=True,
|
|
)
|
|
|
|
@classmethod
|
|
def execute(cls, model) -> io.NodeOutput:
|
|
return io.NodeOutput(model)
|
|
|
|
|
|
class NopExtension(ComfyExtension):
|
|
@override
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
wanBlockSwap
|
|
]
|
|
|
|
async def comfy_entrypoint() -> NopExtension:
|
|
return NopExtension()
|