Commit 0e83c675 authored by AUTOMATIC1111's avatar AUTOMATIC1111

by request: fix tiled vae extension

parent 70c63c12
import torch import torch
import inspect import inspect
import k_diffusion.sampling import k_diffusion.sampling
from modules import sd_samplers_common, sd_samplers_extra, sd_samplers_cfg_denoiser from modules import sd_samplers_common, sd_samplers_extra
from modules.sd_samplers_cfg_denoiser import CFGDenoiser
from modules.shared import opts from modules.shared import opts
import modules.shared as shared import modules.shared as shared
...@@ -62,7 +63,7 @@ class KDiffusionSampler(sd_samplers_common.Sampler): ...@@ -62,7 +63,7 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
denoiser = k_diffusion.external.CompVisVDenoiser if sd_model.parameterization == "v" else k_diffusion.external.CompVisDenoiser denoiser = k_diffusion.external.CompVisVDenoiser if sd_model.parameterization == "v" else k_diffusion.external.CompVisDenoiser
self.model_wrap = denoiser(sd_model, quantize=shared.opts.enable_quantization) self.model_wrap = denoiser(sd_model, quantize=shared.opts.enable_quantization)
self.model_wrap_cfg = sd_samplers_cfg_denoiser.CFGDenoiser(self.model_wrap, self) self.model_wrap_cfg = CFGDenoiser(self.model_wrap, self)
def get_sigmas(self, p, steps): def get_sigmas(self, p, steps):
discard_next_to_last_sigma = self.config is not None and self.config.options.get('discard_next_to_last_sigma', False) discard_next_to_last_sigma = self.config is not None and self.config.options.get('discard_next_to_last_sigma', False)
......
import torch import torch
import inspect import inspect
import sys
from modules import devices, sd_samplers_common, sd_samplers_timesteps_impl from modules import devices, sd_samplers_common, sd_samplers_timesteps_impl
from modules.sd_samplers_cfg_denoiser import CFGDenoiser from modules.sd_samplers_cfg_denoiser import CFGDenoiser
...@@ -145,3 +146,6 @@ class CompVisSampler(sd_samplers_common.Sampler): ...@@ -145,3 +146,6 @@ class CompVisSampler(sd_samplers_common.Sampler):
return samples return samples
sys.modules['modules.sd_samplers_compvis'] = sys.modules[__name__]
VanillaStableDiffusionSampler = CompVisSampler # temp. compatibility with older extensions
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment