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

fix using sd15 model as refiner in second pass

Browse Source
This commit is contained in:
Vladimir Mandic
2024-01-29 13:18:03 -05:00
parent 325b9eb652
commit b949c042b5
4 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+2 -1
View File
@@ -618,7 +618,8 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
)
shared.state.sampling_steps = refiner_args['num_inference_steps']
try:
shared.sd_refiner.register_to_config(requires_aesthetics_score=shared.opts.diffusers_aesthetics_score)
if 'requires_aesthetics_score' in shared.sd_refiner.config:
shared.sd_refiner.register_to_config(requires_aesthetics_score=shared.opts.diffusers_aesthetics_score)
refiner_output = shared.sd_refiner(**refiner_args) # pylint: disable=not-callable
openvino_post_compile(op="refiner")
except AssertionError as e: