lllyasviel commited on
Commit
dbdeecb
·
1 Parent(s): d062d3a
Files changed (1) hide show
  1. 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()