Update app.py
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ def generate_video(positive_prompt, negative_prompt, guidance_scale, bias, step,
|
|
| 62 |
print(img_len)
|
| 63 |
mask = torch.zeros((1, img_len, pos_len+neg_len)).cuda()
|
| 64 |
# mask[:, :, -neg_len:] = -torch.inf # this should be negative
|
| 65 |
-
mask[:, :, -neg_len:] = bias
|
| 66 |
|
| 67 |
for block in pipe.transformer.blocks:
|
| 68 |
block.attn2.processor = WanAttnProcessor2_0(scale=guidance_scale, neg_prompt_length=neg_len, attn_mask=mask)
|
|
|
|
| 62 |
print(img_len)
|
| 63 |
mask = torch.zeros((1, img_len, pos_len+neg_len)).cuda()
|
| 64 |
# mask[:, :, -neg_len:] = -torch.inf # this should be negative
|
| 65 |
+
mask[:, :, -neg_len:] = -bias
|
| 66 |
|
| 67 |
for block in pipe.transformer.blocks:
|
| 68 |
block.attn2.processor = WanAttnProcessor2_0(scale=guidance_scale, neg_prompt_length=neg_len, attn_mask=mask)
|