fix sd35 width/height alignment

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-08-08 13:07:50 -04:00
parent a345ae0e3b
commit 80f603f3ff
3 changed files with 7 additions and 3 deletions
+5 -2
View File
@@ -122,8 +122,11 @@ def task_specific_kwargs(p, model):
'target_subject_category': getattr(p, 'prompt', '').split()[-1],
'output_type': 'pil',
}
# TODO
if model.__class__.__name__ == 'StableDiffusion3Pipeline':
p.width = 16 * (p.width // 16)
p.height = 16 * (p.height // 16)
task_args['width'] = p.width
task_args['height'] = p.height
if debug_enabled:
debug_log(f'Process task specific args: {task_args}')
return task_args