Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
ComfyUI
Commits
b82d67d5
Unverified
Commit
b82d67d5
authored
Jul 01, 2024
by
Peter Crabtree
Committed by
GitHub
Jul 01, 2024
Browse files
Add SamplerEulerAncestralCFG++ custom sampler node (#3901)
(for eta and s_noise)
parent
755c48d7
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
24 additions
and
0 deletions
+24
-0
comfy_extras/nodes_custom_sampler.py
comfy_extras/nodes_custom_sampler.py
+24
-0
No files found.
comfy_extras/nodes_custom_sampler.py
View file @
b82d67d5
...
@@ -293,6 +293,25 @@ class SamplerEulerAncestral:
...
@@ -293,6 +293,25 @@ class SamplerEulerAncestral:
sampler
=
comfy
.
samplers
.
ksampler
(
"euler_ancestral"
,
{
"eta"
:
eta
,
"s_noise"
:
s_noise
})
sampler
=
comfy
.
samplers
.
ksampler
(
"euler_ancestral"
,
{
"eta"
:
eta
,
"s_noise"
:
s_noise
})
return
(
sampler
,
)
return
(
sampler
,
)
class
SamplerEulerAncestralCFGPP
:
@
classmethod
def
INPUT_TYPES
(
s
):
return
{
"required"
:
{
"eta"
:
(
"FLOAT"
,
{
"default"
:
1.0
,
"min"
:
0.0
,
"max"
:
1.0
,
"step"
:
0.01
,
"round"
:
False
}),
"s_noise"
:
(
"FLOAT"
,
{
"default"
:
1.0
,
"min"
:
0.0
,
"max"
:
10.0
,
"step"
:
0.01
,
"round"
:
False
}),
}}
RETURN_TYPES
=
(
"SAMPLER"
,)
CATEGORY
=
"sampling/custom_sampling/samplers"
FUNCTION
=
"get_sampler"
def
get_sampler
(
self
,
eta
,
s_noise
):
sampler
=
comfy
.
samplers
.
ksampler
(
"euler_ancestral_cfg_pp"
,
{
"eta"
:
eta
,
"s_noise"
:
s_noise
})
return
(
sampler
,
)
class
SamplerLMS
:
class
SamplerLMS
:
@
classmethod
@
classmethod
def
INPUT_TYPES
(
s
):
def
INPUT_TYPES
(
s
):
...
@@ -622,6 +641,7 @@ NODE_CLASS_MAPPINGS = {
...
@@ -622,6 +641,7 @@ NODE_CLASS_MAPPINGS = {
"SDTurboScheduler"
:
SDTurboScheduler
,
"SDTurboScheduler"
:
SDTurboScheduler
,
"KSamplerSelect"
:
KSamplerSelect
,
"KSamplerSelect"
:
KSamplerSelect
,
"SamplerEulerAncestral"
:
SamplerEulerAncestral
,
"SamplerEulerAncestral"
:
SamplerEulerAncestral
,
"SamplerEulerAncestralCFGPP"
:
SamplerEulerAncestralCFGPP
,
"SamplerLMS"
:
SamplerLMS
,
"SamplerLMS"
:
SamplerLMS
,
"SamplerDPMPP_3M_SDE"
:
SamplerDPMPP_3M_SDE
,
"SamplerDPMPP_3M_SDE"
:
SamplerDPMPP_3M_SDE
,
"SamplerDPMPP_2M_SDE"
:
SamplerDPMPP_2M_SDE
,
"SamplerDPMPP_2M_SDE"
:
SamplerDPMPP_2M_SDE
,
...
@@ -639,3 +659,7 @@ NODE_CLASS_MAPPINGS = {
...
@@ -639,3 +659,7 @@ NODE_CLASS_MAPPINGS = {
"AddNoise"
:
AddNoise
,
"AddNoise"
:
AddNoise
,
"SamplerCustomAdvanced"
:
SamplerCustomAdvanced
,
"SamplerCustomAdvanced"
:
SamplerCustomAdvanced
,
}
}
NODE_DISPLAY_NAME_MAPPINGS
=
{
"SamplerEulerAncestralCFGPP"
:
"SamplerEulerAncestralCFG++"
,
}
\ No newline at end of file
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment