diff --git a/atoma-inference/src/models/candle/stable_diffusion.rs b/atoma-inference/src/models/candle/stable_diffusion.rs
index a6c1bffe..c16db69e 100644
--- a/atoma-inference/src/models/candle/stable_diffusion.rs
+++ b/atoma-inference/src/models/candle/stable_diffusion.rs
@@ -357,7 +357,6 @@ impl ModelTrait for StableDiffusion {
                     latents.clone()
                 };
 
-                info!("FLAG: {:?}", latent_model_input.shape());
                 let latent_model_input =
                     scheduler.scale_model_input(latent_model_input, timestep)?;
                 let noise_pred =