Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-03 19:40:47 +02:00
Code Issues Packages Projects Releases Wiki Activity

Diffusers subseeds

Browse Source
This commit is contained in:
AI-Casanova
2024-02-17 09:22:16 -06:00
committed by Vladimir Mandic
parent bdce5c1710
commit 50beb2157d
3 changed files with 58 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+5
View File
@@ -170,6 +170,9 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
if shared.opts.diffusers_generator_device == "Unset":
generator_device = None
generator = None
elif getattr(p, "generator", None) is not None:
generator_device = devices.cpu if shared.opts.diffusers_generator_device == "CPU" else shared.device
generator = p.generator
else:
generator_device = devices.cpu if shared.opts.diffusers_generator_device == "CPU" else shared.device
generator = [torch.Generator(generator_device).manual_seed(s) for s in p.seeds]
@@ -222,6 +225,8 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
args['guidance_scale'] = p.cfg_scale
if 'generator' in possible and generator is not None:
args['generator'] = generator
if 'latents' in possible and getattr(p, "init_latent", None) is not None:
args['latents'] = p.init_latent
if 'output_type' in possible:
if hasattr(model, 'vae'):
args['output_type'] = 'np' # only set latent if model has vae