Skip to content

Commit 7697583

Browse files
committed
fix minor bugs
1 parent 1e27df5 commit 7697583

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

null_text_w_ptp.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ def __call__(self, attn, is_cross: bool, place_in_unet: str):
223223
attn = self.forward(attn, is_cross, place_in_unet)
224224
else:
225225
h = attn.shape[0]
226-
attn[h // 2:] = self.forward(attn[h // 2:], is_cross, place_in_unet)
226+
attn[h // 3:] = self.forward(attn[h // 3:], is_cross, place_in_unet)
227227
self.cur_att_layer += 1
228228
if self.cur_att_layer == self.num_att_layers + self.num_uncond_att_layers:
229229
self.cur_att_layer = 0

0 commit comments

Comments
 (0)