Add files via upload

This commit is contained in:
Extraltodeus
2024-08-27 09:33:57 +02:00
committed by GitHub
parent 070455b2a9
commit 6ff12ebed4
3 changed files with 155 additions and 0 deletions
+25
View File
@@ -0,0 +1,25 @@
from .custom_samplers import SamplerDistanceAdvanced
from .presets_to_add import extra_samplers
def add_samplers():
from comfy.samplers import KSampler, k_diffusion_sampling
added = 0
samplers_names = [n for n in extra_samplers][::-1]
for sampler in samplers_names:
if sampler not in KSampler.SAMPLERS:
try:
idx = KSampler.SAMPLERS.index("uni_pc_bh2") # Last item in the samplers list
KSampler.SAMPLERS.insert(idx+1, sampler) # Add our custom samplers
setattr(k_diffusion_sampling, "sample_{}".format(sampler), extra_samplers[sampler])
added += 1
except ValueError as _err:
pass
if added > 0:
import importlib
importlib.reload(k_diffusion_sampling)
add_samplers()
NODE_CLASS_MAPPINGS = {
"SamplerDistance": SamplerDistanceAdvanced,
}