diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py index 6ddd1235..de5f81dd 100644 --- a/toolkit/stable_diffusion_model.py +++ b/toolkit/stable_diffusion_model.py @@ -256,7 +256,7 @@ class StableDiffusion: divisibility = 2 ** (len(self.vae.config['block_out_channels']) - 1) # flux packs this again, - if self.is_flux: + if self.is_flux or self.is_v3: divisibility = divisibility * 2 return divisibility