Merge pull request #171 from blessedcoolant/sampler-bug-fix

Fix sampler changer not working.
This commit is contained in:
blessedcoolant 2022-08-29 13:06:30 +12:00 committed by GitHub
commit 41b26e0520
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -212,7 +212,7 @@ class T2I:
save_original=False, save_original=False,
upscale=None, upscale=None,
variants=None, variants=None,
user_sampler=None, sampler_name=None,
**args, **args,
): # eat up additional cruft ): # eat up additional cruft
""" """
@ -272,8 +272,8 @@ class T2I:
scope = autocast if self.precision == 'autocast' else nullcontext scope = autocast if self.precision == 'autocast' else nullcontext
if user_sampler and (user_sampler != self.sampler_name): if sampler_name and (sampler_name != self.sampler_name):
self.sampler_name = user_sampler self.sampler_name = sampler_name
self._set_sampler() self._set_sampler()
tic = time.time() tic = time.time()