From 3c37c68fe07b03323ac1ebbae4781a286f05dadb Mon Sep 17 00:00:00 2001
From: C43H66N12O12S2 <36072735+C43H66N12O12S2@users.noreply.github.com>
Date: Wed, 14 Sep 2022 18:01:01 +0300
Subject: [PATCH] Construct CompVisDenoiser with quantization

---
 modules/sd_samplers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py
index 7ef507f1..ffb8c2cf 100644
--- 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)
+        self.model_wrap = k_diffusion.external.CompVisDenoiser(sd_model, quantize=True)
         self.funcname = funcname
         self.func = getattr(k_diffusion.sampling, self.funcname)
         self.model_wrap_cfg = CFGDenoiser(self.model_wrap)