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

dev tool (#526)

Browse Source
This commit is contained in:
lllyasviel
2023-10-03 16:08:06 -07:00
committed by GitHub
parent 80632d5eb3
commit 6c56b580d7
5 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/default_pipeline.py
+5 -3
View File
@@ -198,7 +198,7 @@ expansion = FooocusExpansion()
@torch.no_grad()
@torch.inference_mode()
def process_diffusion(positive_cond, negative_cond, steps, switch, width, height, image_seed, callback, sampler_name, latent=None, denoise=1.0, tiled=False, cfg_scale=7.0):
def process_diffusion(positive_cond, negative_cond, steps, switch, width, height, image_seed, callback, sampler_name, scheduler_name, latent=None, denoise=1.0, tiled=False, cfg_scale=7.0):
if latent is None:
empty_latent = core.generate_empty_latent(width=width, height=height, batch_size=1)
else:
@@ -219,7 +219,8 @@ def process_diffusion(positive_cond, negative_cond, steps, switch, width, height
denoise=denoise,
callback_function=callback,
cfg=cfg_scale,
sampler_name=sampler_name
sampler_name=sampler_name,
scheduler=scheduler_name
)
else:
sampled_latent = core.ksampler(
@@ -232,7 +233,8 @@ def process_diffusion(positive_cond, negative_cond, steps, switch, width, height
denoise=denoise,
callback_function=callback,
cfg=cfg_scale,
sampler_name=sampler_name
sampler_name=sampler_name,
scheduler=scheduler_name
)
decoded_latent = core.decode_vae(vae=xl_base_patched.vae, latent_image=sampled_latent, tiled=tiled)