Add a ManualSigmas node.

Can be used to manually set the sigmas for a model.

This node accepts a list of integer and floating point numbers separated
with any non numeric character.
This commit is contained in:
comfyanonymous 2025-12-24 19:04:30 -05:00
parent 4f067b07fb
commit 641f47773e

View File

@ -9,6 +9,7 @@ import comfy.utils
import node_helpers import node_helpers
from typing_extensions import override from typing_extensions import override
from comfy_api.latest import ComfyExtension, io from comfy_api.latest import ComfyExtension, io
import re
class BasicScheduler(io.ComfyNode): class BasicScheduler(io.ComfyNode):
@ -1013,6 +1014,25 @@ class AddNoise(io.ComfyNode):
add_noise = execute add_noise = execute
class ManualSigmas(io.ComfyNode):
@classmethod
def define_schema(cls):
return io.Schema(
node_id="ManualSigmas",
category="_for_testing/custom_sampling",
is_experimental=True,
inputs=[
io.String.Input("sigmas", default="1, 0.5", multiline=False)
],
outputs=[io.Sigmas.Output()]
)
@classmethod
def execute(cls, sigmas) -> io.NodeOutput:
sigmas = re.findall(r"[-+]?(?:\d*\.*\d+)", sigmas)
sigmas = [float(i) for i in sigmas]
sigmas = torch.FloatTensor(sigmas)
return io.NodeOutput(sigmas)
class CustomSamplersExtension(ComfyExtension): class CustomSamplersExtension(ComfyExtension):
@override @override
@ -1052,6 +1072,7 @@ class CustomSamplersExtension(ComfyExtension):
DisableNoise, DisableNoise,
AddNoise, AddNoise,
SamplerCustomAdvanced, SamplerCustomAdvanced,
ManualSigmas,
] ]