fix wan22-5b vae decode

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-10-21 10:52:00 -04:00
parent 4ecdf7bdcf
commit f7d4ffe2f2
4 changed files with 8 additions and 1 deletions
+2
View File
@@ -11,6 +11,8 @@ def load_model(selected: models_def.Model):
if selected is None or selected.te_cls is None or selected.dit_cls is None:
return ''
global loaded_model # pylint: disable=global-statement
if not shared.sd_loaded:
loaded_model = None
if loaded_model == selected.name:
return ''
sd_models.unload_model_weights()
+4
View File
@@ -61,3 +61,7 @@ def set_overrides(p: processing.StableDiffusionProcessingVideo, selected: Model)
if 'WanVACEPipeline' in cls:
if (getattr(p, 'init_images', None) is not None) and (len(p.init_images) > 0):
p.task_args['reference_images'] = p.init_images
# WAN 2.2-5B
if 'WAN 2.2 5B' in selected.name:
print('HERE')
shared.sd_model.vae.disable_tiling()