Allow user to generate images with initial noise as on M1 / mps system

This commit is contained in:
ArDiouscuros
2022-10-07 22:52:14 +02:00
committed by tildebyte
parent 3170c83d8d
commit 0b7ca6a326
5 changed files with 12 additions and 2 deletions

View File

@ -258,6 +258,7 @@ class Generate:
# Set this True to handle KeyboardInterrupt internally
catch_interrupts = False,
hires_fix = False,
use_mps_noise = False,
**args,
): # eat up additional cruft
"""
@ -386,6 +387,7 @@ class Generate:
generator.set_variation(
self.seed, variation_amount, with_variations)
generator.use_mps_noise = use_mps_noise
results = generator.generate(
prompt,
iterations=iterations,