Update divisiblity of SD3

2025-04-02 06:49:06 -06:00 · 2025-04-02 06:49:06 -06:00 · 77763a3e5c
parent a42c5a1de5
commit 77763a3e5c
1 changed files with 1 additions and 1 deletions
--- 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