Enable stable-diffusion 3 on metal. (#2560)

2025-06-17 02:58:50 +00:00 · 2024-10-14 08:59:12 +02:00
parent f553ab5eb4
commit 3d1dc06cdb
4 changed files with 11 additions and 12 deletions
--- a/candle-examples/examples/stable-diffusion-3/sampling.rs
+++ b/candle-examples/examples/stable-diffusion-3/sampling.rs
@ -31,7 +31,7 @@ pub fn euler_sample(
        let timestep = (*s_curr) * 1000.0;
        let noise_pred = mmdit.forward(
            &Tensor::cat(&[x.clone(), x.clone()], 0)?,
-            &Tensor::full(timestep, (2,), x.device())?.contiguous()?,
+            &Tensor::full(timestep as f32, (2,), x.device())?.contiguous()?,
            y,
            context,
        )?;