lllyasviel
commited on
Commit
·
dbdeecb
1
Parent(s):
d062d3a
- modules/core.py +10 -0
modules/core.py
CHANGED
@@ -106,6 +106,11 @@ def close_all_preview():
|
|
106 |
def ksampler(model, positive, negative, latent, seed=None, steps=30, cfg=7.0, sampler_name='dpmpp_2m_sde_gpu',
|
107 |
scheduler='karras', denoise=1.0, disable_noise=False, start_step=None, last_step=None,
|
108 |
force_full_denoise=False):
|
|
|
|
|
|
|
|
|
|
|
109 |
seed = seed if isinstance(seed, int) else random.randint(1, 2 ** 64)
|
110 |
|
111 |
device = comfy.model_management.get_torch_device()
|
@@ -170,6 +175,11 @@ def ksampler_with_refiner(model, positive, negative, refiner, refiner_positive,
|
|
170 |
seed=None, steps=30, refiner_switch_step=20, cfg=7.0, sampler_name='dpmpp_2m_sde_gpu',
|
171 |
scheduler='karras', denoise=1.0, disable_noise=False, start_step=None, last_step=None,
|
172 |
force_full_denoise=False):
|
|
|
|
|
|
|
|
|
|
|
173 |
seed = seed if isinstance(seed, int) else random.randint(1, 2 ** 64)
|
174 |
|
175 |
device = comfy.model_management.get_torch_device()
|
|
|
106 |
def ksampler(model, positive, negative, latent, seed=None, steps=30, cfg=7.0, sampler_name='dpmpp_2m_sde_gpu',
|
107 |
scheduler='karras', denoise=1.0, disable_noise=False, start_step=None, last_step=None,
|
108 |
force_full_denoise=False):
|
109 |
+
# SCHEDULERS = ["normal", "karras", "exponential", "simple", "ddim_uniform"]
|
110 |
+
# SAMPLERS = ["euler", "euler_ancestral", "heun", "dpm_2", "dpm_2_ancestral",
|
111 |
+
# "lms", "dpm_fast", "dpm_adaptive", "dpmpp_2s_ancestral", "dpmpp_sde", "dpmpp_sde_gpu",
|
112 |
+
# "dpmpp_2m", "dpmpp_2m_sde", "dpmpp_2m_sde_gpu", "ddim", "uni_pc", "uni_pc_bh2"]
|
113 |
+
|
114 |
seed = seed if isinstance(seed, int) else random.randint(1, 2 ** 64)
|
115 |
|
116 |
device = comfy.model_management.get_torch_device()
|
|
|
175 |
seed=None, steps=30, refiner_switch_step=20, cfg=7.0, sampler_name='dpmpp_2m_sde_gpu',
|
176 |
scheduler='karras', denoise=1.0, disable_noise=False, start_step=None, last_step=None,
|
177 |
force_full_denoise=False):
|
178 |
+
# SCHEDULERS = ["normal", "karras", "exponential", "simple", "ddim_uniform"]
|
179 |
+
# SAMPLERS = ["euler", "euler_ancestral", "heun", "dpm_2", "dpm_2_ancestral",
|
180 |
+
# "lms", "dpm_fast", "dpm_adaptive", "dpmpp_2s_ancestral", "dpmpp_sde", "dpmpp_sde_gpu",
|
181 |
+
# "dpmpp_2m", "dpmpp_2m_sde", "dpmpp_2m_sde_gpu", "ddim", "uni_pc", "uni_pc_bh2"]
|
182 |
+
|
183 |
seed = seed if isinstance(seed, int) else random.randint(1, 2 ** 64)
|
184 |
|
185 |
device = comfy.model_management.get_torch_device()
|