diff options
| -rw-r--r-- | candle-examples/examples/wuerstchen/main.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-examples/examples/wuerstchen/main.rs b/candle-examples/examples/wuerstchen/main.rs index bce68114..aaa9b78a 100644 --- a/candle-examples/examples/wuerstchen/main.rs +++ b/candle-examples/examples/wuerstchen/main.rs @@ -14,7 +14,7 @@ use candle::{DType, Device, IndexOp, Module, Tensor, D}; use clap::Parser; use tokenizers::Tokenizer; -const PRIOR_GUIDANCE_SCALE: f64 = 8.0; +const PRIOR_GUIDANCE_SCALE: f64 = 4.0; const RESOLUTION_MULTIPLE: f64 = 42.67; const LATENT_DIM_SCALE: f64 = 10.67; const PRIOR_CIN: usize = 16; @@ -354,7 +354,7 @@ fn run(args: Args) -> Result<()> { )?; println!("diffusion process with prior {image_embeddings:?}"); - let scheduler = wuerstchen::ddpm::DDPMWScheduler::new(60, Default::default())?; + let scheduler = wuerstchen::ddpm::DDPMWScheduler::new(12, Default::default())?; let timesteps = scheduler.timesteps(); let timesteps = ×teps[..timesteps.len() - 1]; for (index, &t) in timesteps.iter().enumerate() { |