Skip to content

Commit

Permalink
fix attention masking
Browse files Browse the repository at this point in the history
  • Loading branch information
matt3o committed Feb 10, 2024
1 parent cbcac3d commit 504b6d3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion InstantID.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def __call__(self, n, context_attn2, value_attn2, extra_options):
mask_downsample = mask_downsample.repeat(len(cond_or_uncond), 1, 1)
mask_downsample = mask_downsample.view(mask_downsample.shape[0], -1, 1).repeat(1, 1, out.shape[2])

out_ip = out_ip * mask_downsample
out_iid = out_iid * mask_downsample

out = out + out_iid

Expand Down

0 comments on commit 504b6d3

Please sign in to comment.