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