bug fix

sir1st-inc · sir1st-inc · commit aa6eefc927c1 · 2025-06-19T17:08:01.000+08:00
diff --git a/diffsynth_engine/pipelines/base.py b/diffsynth_engine/pipelines/base.py
@@ -196,7 +196,8 @@ def prepare_latents(
             # if you have any questions about this, please ask @dizhipeng.dzp for more details
             latents = latents * sigmas[0] / ((sigmas[0] ** 2 + 1) ** 0.5)
             init_latents = latents.clone()
-        sigmas, timesteps = sigmas.to(device=self.device), timesteps.to(self.device)
+        sigmas, timesteps = sigmas.to(device=self.device, dtype=self.dtype), timesteps.to(device=self.device, dtype=self.dtype)
+        init_latents, latents = init_latents.to(device=self.device, dtype=self.dtype), latents.to(device=self.device, dtype=self.dtype)
         return init_latents, latents, sigmas, timesteps
 
     def eval(self):
diff --git a/diffsynth_engine/pipelines/flux_image.py b/diffsynth_engine/pipelines/flux_image.py
@@ -676,7 +676,8 @@ def prepare_latents(
                 num_inference_steps, mu=mu, sigma_min=1 / num_inference_steps, sigma_max=1.0
             )
             init_latents = latents.clone()
-        sigmas, timesteps = sigmas.to(device=self.device), timesteps.to(self.device)
+        sigmas, timesteps = sigmas.to(device=self.device, dtype=self.dtype), timesteps.to(device=self.device, dtype=self.dtype)
+        init_latents, latents = init_latents.to(device=self.device, dtype=self.dtype), latents.to(device=self.device, dtype=self.dtype)
         return init_latents, latents, sigmas, timesteps
 
     def prepare_masked_latent(self, image: Image.Image, mask: Image.Image | None, height: int, width: int):

Original file line number	Diff line number	Diff line change
`@@ -676,7 +676,8 @@ def prepare_latents(`
`676`	`676`	`num_inference_steps, mu=mu, sigma_min=1 / num_inference_steps, sigma_max=1.0`
`677`	`677`	`)`
`678`	`678`	`init_latents = latents.clone()`
`679`		`- sigmas, timesteps = sigmas.to(device=self.device), timesteps.to(self.device)`
	`679`	`+ sigmas, timesteps = sigmas.to(device=self.device, dtype=self.dtype), timesteps.to(device=self.device, dtype=self.dtype)`
	`680`	`+ init_latents, latents = init_latents.to(device=self.device, dtype=self.dtype), latents.to(device=self.device, dtype=self.dtype)`
`680`	`681`	`return init_latents, latents, sigmas, timesteps`
`681`	`682`
`682`	`683`	`def prepare_masked_latent(self, image: Image.Image, mask: Image.Image \| None, height: int, width: int):`