Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-14 02:28:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

update svd

Browse Source
This commit is contained in:
Vladimir Mandic
2024-03-13 09:18:07 -04:00
parent a4c6257124
commit 50b09082a9
6 changed files with 81 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+3 -2
View File
@@ -416,6 +416,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
p.extra_generation_params['Pipeline'] = shared.sd_model.__class__.__name__
if shared.opts.scheduler_eta is not None and shared.opts.scheduler_eta > 0 and shared.opts.scheduler_eta < 1:
p.extra_generation_params["Sampler Eta"] = shared.opts.scheduler_eta
output = None
try:
t0 = time.time()
sd_models_compile.check_deepcache(enable=True)
@@ -450,7 +451,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
if hasattr(shared.sd_model, 'embedding_db') and len(shared.sd_model.embedding_db.embeddings_used) > 0: # register used embeddings
p.extra_generation_params['Embeddings'] = ', '.join(shared.sd_model.embedding_db.embeddings_used)
if hasattr(p, 'task_args') and p.task_args.get('image', None) is not None: # replace input with output so it can be used by hires/refine
if hasattr(p, 'task_args') and p.task_args.get('image', None) is not None and output is not None: # replace input with output so it can be used by hires/refine
p.task_args['image'] = output.images
shared.state.nextjob()
@@ -476,7 +477,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
save_intermediate(latents=output.images, suffix="-before-hires")
shared.state.job = 'upscale'
output.images = resize_hires(p, latents=output.images)
if hasattr(p, 'task_args') and p.task_args.get('image', None) is not None: # replace input with output so it can be used by hires/refine
if hasattr(p, 'task_args') and p.task_args.get('image', None) is not None and output is not None: # replace input with output so it can be used by hires/refine
p.task_args['image'] = output.images
sd_hijack_hypertile.hypertile_set(p, hr=True)