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

fix tgate apply/unapply

Browse Source
This commit is contained in:
Vladimir Mandic
2024-06-04 07:12:12 -04:00
parent 901784f3b5
commit f84705d079
3 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+2 -1
View File
@@ -116,7 +116,8 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
hidiffusion.apply_hidiffusion(p, shared.sd_model_type)
# if 'image' in base_args:
# base_args['image'] = set_latents(p)
if hasattr(shared.sd_model, 'tgate'):
if hasattr(shared.sd_model, 'tgate') and getattr(p, 'gate_step', -1) > 0:
base_args['gate_step'] = p.gate_step
output = shared.sd_model.tgate(**base_args) # pylint: disable=not-callable
else:
output = shared.sd_model(**base_args)