Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

don't suppress exceptions when doing cross-attention control

Browse Source
This commit is contained in:
damian0815 2022-11-08 14:28:38 +01:00 committed by Lincoln Stein
parent 71bbfe4a1a
commit b17ca0a5e7
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ldm/models/diffusion/shared_invokeai_diffusion.py
View File

@ -151,10 +151,12 @@ class InvokeAIDiffuserComponent:
context.request_apply_saved_attention_maps(ca_type) context.request_apply_saved_attention_maps(ca_type)
edited_conditioning = self.conditioning.cross_attention_control_args.edited_conditioning edited_conditioning = self.conditioning.cross_attention_control_args.edited_conditioning
conditioned_next_x = self.model_forward_callback(x, sigma, edited_conditioning) conditioned_next_x = self.model_forward_callback(x, sigma, edited_conditioning)
finally:
context.clear_requests(cleanup=True) context.clear_requests(cleanup=True)
except:
context.clear_requests(cleanup=True)
raise
return unconditioned_next_x, conditioned_next_x return unconditioned_next_x, conditioned_next_x
def estimate_percent_through(self, step_index, sigma): def estimate_percent_through(self, step_index, sigma):