Respect quantization setting

3 years ago · 81baaff9b3
parent 829d7150bc
commit 81baaff9b3
1 changed files with 1 additions and 1 deletions
--- a/modules/sd_samplers.py
+++ b/modules/sd_samplers.py
@ -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)
+        self.model_wrap = k_diffusion.external.CompVisDenoiser(sd_model, quantize=True if shared.opts.enable_quantization else False)
        self.funcname = funcname
        self.func = getattr(k_diffusion.sampling, self.funcname)
        self.model_wrap_cfg = CFGDenoiser(self.model_wrap)