mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
fix wan22-5b vae decode
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -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()
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user