cleanup reference models

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-08-08 14:07:33 -04:00
parent 80f603f3ff
commit 00e34ce0d3
7 changed files with 38 additions and 56 deletions
+1 -1
View File
@@ -122,7 +122,7 @@ def task_specific_kwargs(p, model):
'target_subject_category': getattr(p, 'prompt', '').split()[-1],
'output_type': 'pil',
}
if model.__class__.__name__ == 'StableDiffusion3Pipeline':
if model.__class__.__name__ in ['StableDiffusion3Pipeline', 'WanPipeline']:
p.width = 16 * (p.width // 16)
p.height = 16 * (p.height // 16)
task_args['width'] = p.width