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

add node ConditioningTimestepInterpolate

Browse Source
This commit is contained in:
Silver 2024-09-16 17:29:12 +02:00
parent 56e8f5e4fd
commit 7a47150e8e
1 changed files with 43 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
nodes.py
View File

@ -264,6 +264,48 @@ class ConditioningSetTimestepRange:
"end_percent": end}) "end_percent": end})
return (c, ) return (c, )
class ConditioningTimestepInterpolate:
@classmethod
def INPUT_TYPES(s):
return {"required": {"conditioning_1": ("CONDITIONING", ),
"conditioning_2": ("CONDITIONING", ),
"interval_range": ("FLOAT", {"default": 0.25, "min": 0.001, "max": 0.5, "step": 0.001})
}}
RETURN_TYPES = ("CONDITIONING",)
FUNCTION = "interpolate_conds"
CATEGORY = "advanced/conditioning"
def interpolate_conds(self, conditioning_1, conditioning_2, interval_range):
num_intervals = int(1 / interval_range)
conditioning_1_intervals = []
conditioning_2_intervals = []
for i in range(num_intervals):
start = i * interval_range
end = (i + 1) * interval_range
if i % 2 == 0:
conditioning_1_intervals.append(node_helpers.conditioning_set_values(conditioning_1, {"start_percent": start, "end_percent": end}))
else:
conditioning_2_intervals.append(node_helpers.conditioning_set_values(conditioning_2, {"start_percent": start, "end_percent": end}))
# Combine intervals for conditioning_1 and conditioning_2
conditioning_1_intervals_combined = conditioning_1_intervals[0]
for i in range(1, len(conditioning_1_intervals)):
conditioning_1_intervals_combined = (conditioning_1_intervals_combined + conditioning_1_intervals[i])
conditioning_2_intervals_combined = conditioning_2_intervals[0]
for i in range(1, len(conditioning_2_intervals)):
conditioning_2_intervals_combined = (conditioning_2_intervals_combined + conditioning_2_intervals[i])
c = (conditioning_1_intervals_combined + conditioning_2_intervals_combined)
return (c, )
class VAEDecode: class VAEDecode:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -1888,6 +1930,7 @@ NODE_CLASS_MAPPINGS = {
"ConditioningZeroOut": ConditioningZeroOut, "ConditioningZeroOut": ConditioningZeroOut,
"ConditioningSetTimestepRange": ConditioningSetTimestepRange, "ConditioningSetTimestepRange": ConditioningSetTimestepRange,
"LoraLoaderModelOnly": LoraLoaderModelOnly, "LoraLoaderModelOnly": LoraLoaderModelOnly,
"ConditioningTimestepInterpolate": ConditioningTimestepInterpolate,
} }
NODE_DISPLAY_NAME_MAPPINGS = { NODE_DISPLAY_NAME_MAPPINGS = {