check in rest of sampler fix

This commit is contained in:
Benjamin Warner 2022-08-28 19:53:25 -05:00
parent 529040708b
commit 594c0eeb8c

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()