Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lvmin
2023-08-10 10:19:28 -07:00
parent 34b1e3a3c5
commit 723a4b60d9
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/core.py
+5 -4
View File
@@ -11,8 +11,7 @@ import comfy.utils
from comfy.sd import load_checkpoint_guess_config from comfy.sd import load_checkpoint_guess_config
from nodes import VAEDecode, EmptyLatentImage, CLIPTextEncode from nodes import VAEDecode, EmptyLatentImage, CLIPTextEncode
from comfy.sample import prepare_mask, broadcast_cond, load_additional_models, cleanup_additional_models from comfy.sample import prepare_mask, broadcast_cond, load_additional_models, cleanup_additional_models
from modules.samplers_advanced import KSamplerAdvanced from modules.samplers_advanced import KSampler, KSamplerWithRefiner
opCLIPTextEncode = CLIPTextEncode() opCLIPTextEncode = CLIPTextEncode()
opEmptyLatentImage = EmptyLatentImage() opEmptyLatentImage = EmptyLatentImage()
@@ -81,7 +80,9 @@ def close_all_preview():
@torch.no_grad() @torch.no_grad()
def ksampler(model, positive, negative, latent, seed=None, steps=30, cfg=9.0, sampler_name='dpmpp_2m_sde', scheduler='karras', denoise=1.0, disable_noise=False, start_step=None, last_step=None, force_full_denoise=False): def ksampler(model, positive, negative, latent, seed=None, steps=30, cfg=9.0, sampler_name='dpmpp_2m_sde',
scheduler='karras', denoise=1.0, disable_noise=False, start_step=None, last_step=None,
force_full_denoise=False):
seed = seed if isinstance(seed, int) else random.randint(1, 2 ** 64) seed = seed if isinstance(seed, int) else random.randint(1, 2 ** 64)
device = comfy.model_management.get_torch_device() device = comfy.model_management.get_torch_device()
@@ -123,7 +124,7 @@ def ksampler(model, positive, negative, latent, seed=None, steps=30, cfg=9.0, sa
models = load_additional_models(positive, negative, model.model_dtype()) models = load_additional_models(positive, negative, model.model_dtype())
sampler = KSamplerAdvanced(real_model, steps=steps, device=device, sampler=sampler_name, scheduler=scheduler, sampler = KSampler(real_model, steps=steps, device=device, sampler=sampler_name, scheduler=scheduler,
denoise=denoise, model_options=model.model_options) denoise=denoise, model_options=model.model_options)
samples = sampler.sample(noise, positive_copy, negative_copy, cfg=cfg, latent_image=latent_image, samples = sampler.sample(noise, positive_copy, negative_copy, cfg=cfg, latent_image=latent_image,
modules/samplers_advanced.py
+1 -1
View File
@@ -1,7 +1,7 @@
from comfy.samplers import * from comfy.samplers import *
class KSamplerAdvanced: class KSamplerWithRefiner:
SCHEDULERS = ["normal", "karras", "exponential", "simple", "ddim_uniform"] SCHEDULERS = ["normal", "karras", "exponential", "simple", "ddim_uniform"]
SAMPLERS = ["euler", "euler_ancestral", "heun", "dpm_2", "dpm_2_ancestral", SAMPLERS = ["euler", "euler_ancestral", "heun", "dpm_2", "dpm_2_ancestral",
"lms", "dpm_fast", "dpm_adaptive", "dpmpp_2s_ancestral", "dpmpp_sde", "dpmpp_sde_gpu", "lms", "dpm_fast", "dpm_adaptive", "dpmpp_2s_ancestral", "dpmpp_sde", "dpmpp_sde_gpu",