diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-09-20 09:26:24 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-20 09:26:24 +0100 |
| commit | c0b49d5a50a23e7fc466deb8dd52e8a2b8266fa4 (patch) | |
| tree | e56c9ebec8d52d464929f3f3ef8b17225237ac84 | |
| parent | 098dd0d1e9cc2b1ca902e4e0d77a9abe3de72a9c (diff) | |
| download | candle-c0b49d5a50a23e7fc466deb8dd52e8a2b8266fa4.tar.gz candle-c0b49d5a50a23e7fc466deb8dd52e8a2b8266fa4.tar.bz2 candle-c0b49d5a50a23e7fc466deb8dd52e8a2b8266fa4.zip | |
Wuerstchen parameter tweaks. (#907)
| -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() { |