Add handles_tiling, format_encoded, audio_sample_rate to CustomVAE init

Signed-off-by: Sai Asish Y <say.apm35@gmail.com>
This commit is contained in:
Sai Asish Y
2026-08-23 12:35:34 -07:00
parent b6a20ae18f
commit a11916790e
+3
View File
@@ -54,6 +54,9 @@ class CustomVAE(VAE):
self.upscale_index_formula = None self.upscale_index_formula = None
self.extra_1d_channel = None self.extra_1d_channel = None
self.crop_input = True self.crop_input = True
self.handles_tiling = False
self.format_encoded = None
self.audio_sample_rate = 44100
if config is None: if config is None:
if "decoder.mid.block_1.mix_factor" in sd: if "decoder.mid.block_1.mix_factor" in sd: