From c0b49d5a50a23e7fc466deb8dd52e8a2b8266fa4 Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Wed, 20 Sep 2023 09:26:24 +0100 Subject: [PATCH] Wuerstchen parameter tweaks. (#907) --- candle-examples/examples/wuerstchen/main.rs | 4 ++-- 1 file 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() {