From ea3c846eae1aa399d138aba0dff77a235d87fcc5 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Thu, 30 Nov 2023 10:14:39 -0600 Subject: [PATCH] Brought back ControlNetLoaderAdvanced/Diff timestep_keyframe names to avoid node name cutoff --- control/nodes.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/control/nodes.py b/control/nodes.py index 0ca0740..3794ac7 100644 --- a/control/nodes.py +++ b/control/nodes.py @@ -71,7 +71,7 @@ class ControlNetLoaderAdvanced: "control_net_name": (folder_paths.get_filename_list("controlnet"), ), }, "optional": { - "timestep_kf": ("TIMESTEP_KEYFRAME", ), + "timestep_keyframe": ("TIMESTEP_KEYFRAME", ), } } @@ -81,9 +81,8 @@ class ControlNetLoaderAdvanced: CATEGORY = "Adv-ControlNet 🛂🅐🅒🅝" def load_controlnet(self, control_net_name, - timestep_kf: TimestepKeyframeGroup=None, timestep_keyframe: TimestepKeyframeGroup=None # old name + timestep_keyframe: TimestepKeyframeGroup=None ): - timestep_keyframe = timestep_keyframe if timestep_keyframe else timestep_kf controlnet_path = folder_paths.get_full_path("controlnet", control_net_name) controlnet = load_controlnet(controlnet_path, timestep_keyframe) return (controlnet,) @@ -98,7 +97,7 @@ class DiffControlNetLoaderAdvanced: "control_net_name": (folder_paths.get_filename_list("controlnet"), ) }, "optional": { - "timestep_kf": ("TIMESTEP_KEYFRAME", ), + "timestep_keyframe": ("TIMESTEP_KEYFRAME", ), } } @@ -108,9 +107,8 @@ class DiffControlNetLoaderAdvanced: CATEGORY = "Adv-ControlNet 🛂🅐🅒🅝" def load_controlnet(self, control_net_name, model, - timestep_kf: TimestepKeyframeGroup=None, timestep_keyframe: TimestepKeyframeGroup=None # old name + timestep_keyframe: TimestepKeyframeGroup=None ): - timestep_keyframe = timestep_keyframe if timestep_keyframe else timestep_kf controlnet_path = folder_paths.get_full_path("controlnet", control_net_name) controlnet = load_controlnet(controlnet_path, timestep_keyframe, model) if is_advanced_controlnet(controlnet):