Commit 81baaff9 authored by C43H66N12O12S2's avatar C43H66N12O12S2 Committed by AUTOMATIC1111

Respect quantization setting

parent 829d7150
......@@ -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)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment