quantization off by default

This commit is contained in:
AUTOMATIC
2022-09-15 14:55:38 +03:00
parent 81baaff9b3
commit d7f7378828
2 changed files with 2 additions and 2 deletions

View File

@@ -173,7 +173,7 @@ def extended_trange(count, *args, **kwargs):
class KDiffusionSampler:
def __init__(self, funcname, sd_model):
self.model_wrap = k_diffusion.external.CompVisDenoiser(sd_model, quantize=True if shared.opts.enable_quantization else False)
self.model_wrap = k_diffusion.external.CompVisDenoiser(sd_model, quantize=shared.opts.enable_quantization)
self.funcname = funcname
self.func = getattr(k_diffusion.sampling, self.funcname)
self.model_wrap_cfg = CFGDenoiser(self.model_wrap)