Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-26 23:20:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

restore pipe at the end of ops

Browse Source
This commit is contained in:
Vladimir Mandic
2024-01-30 13:37:32 -05:00
parent 5faa144a29
commit 1c40ebf32b
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+5
View File
@@ -17,6 +17,7 @@ debug_steps('Trace: STEPS')
def process_diffusers(p: processing.StableDiffusionProcessing):
debug(f'Process diffusers args: {vars(p)}')
orig_pipeline = shared.sd_model
results = []
def is_txt2img():
@@ -409,6 +410,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
p.init_images.append(p.init_images[-1])
if shared.state.interrupted or shared.state.skipped:
shared.sd_model = orig_pipeline
return results
if shared.opts.diffusers_move_base and not getattr(shared.sd_model, 'has_accelerate', False):
@@ -520,6 +522,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
shared.state.nextjob()
if shared.state.interrupted or shared.state.skipped:
shared.sd_model = orig_pipeline
return results
# optional hires pass
@@ -583,6 +586,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
update_sampler(shared.sd_refiner, second_pass=True)
if shared.state.interrupted or shared.state.skipped:
shared.sd_model = orig_pipeline
return results
if shared.opts.diffusers_move_refiner and not getattr(shared.sd_refiner, 'has_accelerate', False):
@@ -656,4 +660,5 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
shared.log.warning('Processing returned no results')
results = []
shared.sd_model = orig_pipeline
return results