Skip to content

Commit 81baaff

Browse files
C43H66N12O12S2AUTOMATIC1111
authored andcommitted
Respect quantization setting
1 parent 829d715 commit 81baaff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/sd_samplers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,7 @@ def extended_trange(count, *args, **kwargs):
173173

174174
class KDiffusionSampler:
175175
def __init__(self, funcname, sd_model):
176-
self.model_wrap = k_diffusion.external.CompVisDenoiser(sd_model, quantize=True)
176+
self.model_wrap = k_diffusion.external.CompVisDenoiser(sd_model, quantize=True if shared.opts.enable_quantization else False)
177177
self.funcname = funcname
178178
self.func = getattr(k_diffusion.sampling, self.funcname)
179179
self.model_wrap_cfg = CFGDenoiser(self.model_wrap)

0 commit comments

Comments
 (0)