Skip to content

Commit 90c73be

Browse files
qzzz95Glaceon-Hyy
andauthored
fix mask dtype differ from latent (#195)
Co-authored-by: yueyang.hyy <[email protected]>
1 parent 96f5d4e commit 90c73be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

diffsynth_engine/pipelines/flux_image.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -830,7 +830,7 @@ def prepare_masked_latent(self, image: Image.Image, mask: Image.Image | None, he
830830
masked_image = image.clone()
831831
masked_image[(mask > 0.5).repeat(1, 3, 1, 1)] = -1
832832
latent = self.encode_image(masked_image)
833-
mask = torch.nn.functional.interpolate(mask, size=(latent.shape[2], latent.shape[3]))
833+
mask = torch.nn.functional.interpolate(mask, size=(latent.shape[2], latent.shape[3])).to(latent.dtype)
834834
mask = 1 - mask
835835
latent = torch.cat([latent, mask], dim=1)
836836
elif self.config.control_type == ControlType.bfl_fill:

0 commit comments

Comments
 (0)