Fix sampler changer not working.

This commit is contained in:
blessedcoolant 2022-08-29 11:26:19 +12:00
parent 373a2d9c32
commit 8a517fdc17

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