Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Construct CompVisDenoiser with quantization

This commit is contained in:
C43H66N12O12S2 2022-09-14 18:01:01 +03:00 committed by AUTOMATIC1111
parent cdaab233c4
commit 3c37c68fe0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers.py
View File

@ -173,7 +173,7 @@ def extended_trange(count, *args, **kwargs):
class KDiffusionSampler: class KDiffusionSampler:
def __init__(self, funcname, sd_model): 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.funcname = funcname
self.func = getattr(k_diffusion.sampling, self.funcname) self.func = getattr(k_diffusion.sampling, self.funcname)
self.model_wrap_cfg = CFGDenoiser(self.model_wrap) self.model_wrap_cfg = CFGDenoiser(self.model_wrap)