Update divisiblity of SD3
This commit is contained in:
1 parent
a42c5a1de5
commit
77763a3e5c
1 file changed
+1
-1
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user