comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-13 09:10:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove curve to sigmas node

Browse Source
This commit is contained in:
Terry Jia
2026-03-12 22:35:23 -04:00
parent 04d29497d5
commit 98d9b7df2f
1 changed files with 0 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
comfy_extras/nodes_custom_sampler.py
View File

@@ -1049,48 +1049,6 @@ class ManualSigmas(io.ComfyNode):
sigmas = torch.FloatTensor(sigmas)
return io.NodeOutput(sigmas)
class CurveToSigmas(io.ComfyNode):
@classmethod
def define_schema(cls):
return io.Schema(
node_id="CurveToSigmas",
display_name="Curve to Sigmas",
category="sampling/custom_sampling/schedulers",
inputs=[
io.Curve.Input("curve", default=[[0.0, 1.0], [1.0, 0.0]]),
io.Model.Input("model", optional=True),
io.Int.Input("steps", default=20, min=1, max=10000),
io.Float.Input("sigma_max", default=14.614642, min=0.0, max=5000.0, step=0.01, round=False, optional=True),
],
outputs=[io.Sigmas.Output()]
)
@classmethod
def execute(cls, curve, steps, sigma_max=14.614642, model=None) -> io.NodeOutput:
points = sorted(curve, key=lambda p: p[0])
model_sampling = model.get_model_object("model_sampling") if model is not None else None
sigmas = []
for i in range(steps + 1):
t = i / steps
y = points[0][1] if t < points[0][0] else points[-1][1]
for j in range(len(points) - 1):
if points[j][0] <= t <= points[j + 1][0]:
x0, y0 = points[j]
x1, y1 = points[j + 1]
y = y0 if x1 == x0 else y0 + (y1 - y0) * (t - x0) / (x1 - x0)
break
if model_sampling is not None:
sigmas.append(float(model_sampling.percent_to_sigma(1.0 - y)))
else:
sigmas.append(y * sigma_max)
sigmas[-1] = 0.0
return io.NodeOutput(torch.FloatTensor(sigmas))
get_sigmas = execute
class CustomSamplersExtension(ComfyExtension):
@override
async def get_node_list(self) -> list[type[io.ComfyNode]]:
@@ -1130,7 +1088,6 @@ class CustomSamplersExtension(ComfyExtension):
AddNoise,
SamplerCustomAdvanced,
ManualSigmas,
CurveToSigmas,
]