From 3769d8bf71d6544131a8dd6ebed8e4f3732156e6 Mon Sep 17 00:00:00 2001
From: laurent <laurent.mazare@gmail.com>
Date: Tue, 19 Sep 2023 10:26:18 +0100
Subject: [PATCH] Print the prior shape.

---
 candle-examples/examples/wuerstchen/main.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/candle-examples/examples/wuerstchen/main.rs b/candle-examples/examples/wuerstchen/main.rs
index ac0526af..46403e47 100644
--- a/candle-examples/examples/wuerstchen/main.rs
+++ b/candle-examples/examples/wuerstchen/main.rs
@@ -350,7 +350,7 @@ fn run(args: Args) -> Result<()> {
             &device,
         )?;
 
-        println!("diffusion process");
+        println!("diffusion process with prior {effnet:?}");
         let scheduler = wuerstchen::ddpm::DDPMWScheduler::new(60, Default::default())?;
         let timesteps = scheduler.timesteps();
         for (index, &t) in timesteps.iter().enumerate() {