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

lint cleanup

Browse Source
This commit is contained in:
Vladimir Mandic
2023-10-16 12:07:37 -04:00
parent 44f817acc6
commit 21595ee9e7
13 changed files with 56 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+6 -5
View File
@@ -201,7 +201,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
return task_args
def set_pipeline_args(model, prompts: list, negative_prompts: list, prompts_2: typing.Optional[list]=None, negative_prompts_2: typing.Optional[list]=None, desc:str='', **kwargs):
if hasattr(model, "set_progress_bar_config"):
model.set_progress_bar_config(bar_format='Progress {rate_fmt}{postfix} {bar} {percentage:3.0f}% {n_fmt}/{total_fmt} {elapsed} {remaining} ' + '\x1b[38;5;71m' + desc, ncols=80, colour='#327fba')
args = {}
@@ -323,8 +323,9 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
sampler = sd_samplers.all_samplers_map.get(p.sampler_name, None)
if sampler is None:
sampler = sd_samplers.all_samplers_map.get("UniPC")
sd_samplers.create_sampler(sampler.name, shared.sd_model) # TODO(Patrick): For wrapped pipelines this is currently a no-op
# p.extra_generation_params['Sampler options'] = '' # TODO sampler_options
sd_samplers.create_sampler(sampler.name, shared.sd_model)
# TODO extra_generation_params add sampler options
# p.extra_generation_params['Sampler options'] = ''
p.extra_generation_params['Pipeline'] = shared.sd_model.__class__.__name__
@@ -406,7 +407,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
sampler = sd_samplers.all_samplers_map.get(p.latent_sampler, None)
if sampler is None:
sampler = sd_samplers.all_samplers_map.get("UniPC")
sd_samplers.create_sampler(sampler.name, shared.sd_model) # TODO(Patrick): For wrapped pipelines this is currently a no-op
sd_samplers.create_sampler(sampler.name, shared.sd_model)
hires_args = set_pipeline_args(
model=shared.sd_model,
prompts=[p.refiner_prompt] if len(p.refiner_prompt) > 0 else prompts,
@@ -442,7 +443,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
sampler = sd_samplers.all_samplers_map.get(p.latent_sampler, None)
if sampler is None:
sampler = sd_samplers.all_samplers_map.get("UniPC")
sd_samplers.create_sampler(sampler.name, shared.sd_refiner) # TODO(Patrick): For wrapped pipelines this is currently a no-op
sd_samplers.create_sampler(sampler.name, shared.sd_refiner)
if shared.state.interrupted or shared.state.skipped:
return results