correct minor bugs

atoma-network · Apr 6, 2024 · b5fb727 · b5fb727
1 parent ceb52cf
commit b5fb727
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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 =