Merge pull request #169 from modelscope/Artiprocher-dev

fix bug
2026-03-19 06:48:12 +00:00 · 2024-08-23 09:28:46 +08:00
parent ae5e4b67dc 03d91f6618
commit 0488f90c8f
1 changed files with 1 additions and 1 deletions
--- a/diffsynth/pipelines/base.py
+++ b/diffsynth/pipelines/base.py
@@ -37,7 +37,7 @@ class BasePipeline(torch.nn.Module):
        height, width = value.shape[-2:]
        weight = torch.ones_like(value)
        for latent, mask, scale in zip(latents, masks, scales):
-            mask = self.preprocess_image(mask.resize((height, width))).mean(dim=1, keepdim=True) > 0
+            mask = self.preprocess_image(mask.resize((width, height))).mean(dim=1, keepdim=True) > 0
            mask = mask.repeat(1, latent.shape[1], 1, 1)
            value[mask] += latent[mask] * scale
            weight[mask] += scale