mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-27 23:00:20 +08:00
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:
parent
4f067b07fb
commit
641f47773e
@ -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,
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user