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