xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 01:36:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Change manual context window nodes names/ids

Browse Source
This commit is contained in:
Jedrzej Kosinski 2025-08-07 17:41:27 -07:00
parent 3a8be588a6
commit 1cc8acadc6
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
comfy_extras/nodes_context_windows.py
View File

@ -8,10 +8,10 @@ class ContextWindowsManualNode(io.ComfyNode):
@classmethod @classmethod
def define_schema(cls) -> io.Schema: def define_schema(cls) -> io.Schema:
return io.Schema( return io.Schema(
node_id="ContexWindowsTest", node_id="ContextWindowsManual",
display_name="Context Windows Test", display_name="Context Windows (Manual)",
category="context", category="context",
description="Test node for context windows", description="Manually set context windows.",
inputs=[ inputs=[
io.Model.Input("model", tooltip="The model to apply context windows to during sampling."), io.Model.Input("model", tooltip="The model to apply context windows to during sampling."),
io.Int.Input("context_length", min=1, default=1, tooltip="The length of the context window."), io.Int.Input("context_length", min=1, default=1, tooltip="The length of the context window."),
@ -48,13 +48,13 @@ class ContextWindowsManualNode(io.ComfyNode):
comfy.context_windows.create_prepare_sampling_wrapper(model) comfy.context_windows.create_prepare_sampling_wrapper(model)
return io.NodeOutput(model) return io.NodeOutput(model)
class WanContextWindowsNode(ContextWindowsManualNode): class WanContextWindowsManualNode(ContextWindowsManualNode):
@classmethod @classmethod
def define_schema(cls) -> io.Schema: def define_schema(cls) -> io.Schema:
schema = super().define_schema() schema = super().define_schema()
schema.node_id = "WanContextWindowsTest" schema.node_id = "WanContextWindowsManual"
schema.display_name = "Wan Context Windows Test" schema.display_name = "WAN Context Windows (Manual)"
schema.description = "Test node for context windows (WAN)" schema.description = "Manually set context windows for WAN-like models (dim=2)."
# remove dim input; will always be 2 # remove dim input; will always be 2
schema.inputs = [x for x in schema.inputs if x.id != "dim"] schema.inputs = [x for x in schema.inputs if x.id != "dim"]
# replace context_length input; should be in steps of 4 # replace context_length input; should be in steps of 4
@ -79,7 +79,7 @@ class ContextWindowsExtension(ComfyExtension):
async def get_node_list(self) -> list[type[io.ComfyNode]]: async def get_node_list(self) -> list[type[io.ComfyNode]]:
return [ return [
ContextWindowsManualNode, ContextWindowsManualNode,
WanContextWindowsNode, WanContextWindowsManualNode,
] ]
def comfy_entrypoint(): def comfy_entrypoint():