diff options
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index c042c5c3..df3a6fe8 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -185,7 +185,7 @@ def extended_trange(count, *args, **kwargs): class KDiffusionSampler:
def __init__(self, funcname, sd_model):
- self.model_wrap = k_diffusion.external.CompVisDenoiser(sd_model)
+ 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)
|