diff options
| author | C43H66N12O12S2 <36072735+C43H66N12O12S2@users.noreply.github.com> | 2022-09-15 07:52:11 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-09-15 11:54:52 +0000 |
| commit | 81baaff9b36bef62f9a083243dc038e8ca769566 (patch) | |
| tree | cfe94155cf7f981cef9b0880acf59bba3bda63a4 /modules/sd_samplers.py | |
| parent | 829d7150bc678b37ecb96e50650755323bbf1c59 (diff) | |
| download | stable-diffusion-webui-gfx803-81baaff9b36bef62f9a083243dc038e8ca769566.tar.gz stable-diffusion-webui-gfx803-81baaff9b36bef62f9a083243dc038e8ca769566.tar.bz2 stable-diffusion-webui-gfx803-81baaff9b36bef62f9a083243dc038e8ca769566.zip | |
Respect quantization setting
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 ffb8c2cf..e1c913b3 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, 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)
|