Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-29 16:41:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix blipdiffusion

Browse Source
This commit is contained in:
Vladimir Mandic
2024-01-27 08:53:17 -05:00
parent 0bfb17ba72
commit e6a1946e06
3 changed files with 19 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+10
View File
@@ -189,6 +189,16 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
'width': p.width if hasattr(p, 'width') else None,
'height': p.height if hasattr(p, 'height') else None,
}
if model.__class__.__name__ == 'BlipDiffusionPipeline':
if len(getattr(p, 'init_images', [])) == 0:
shared.log.error('BLiP diffusion requires init image')
return task_args
task_args = {
'reference_image': p.init_images[0],
'source_subject_category': getattr(p, 'negative_prompt', '').split()[-1],
'target_subject_category': getattr(p, 'prompt', '').split()[-1],
'output_type': 'pil',
}
debug(f'Diffusers task specific args: {task_args}')
return task_args