huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Wuerstchen parameter tweaks. (#907)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-20 09:26:24 +01:00
committed by GitHub
parent 098dd0d1e9
commit c0b49d5a50
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-examples/examples/wuerstchen/main.rs
View File

@ -14,7 +14,7 @@ use candle::{DType, Device, IndexOp, Module, Tensor, D};
use clap::Parser; use clap::Parser;
use tokenizers::Tokenizer; use tokenizers::Tokenizer;
const PRIOR_GUIDANCE_SCALE: f64 = 8.0; const PRIOR_GUIDANCE_SCALE: f64 = 4.0;
const RESOLUTION_MULTIPLE: f64 = 42.67; const RESOLUTION_MULTIPLE: f64 = 42.67;
const LATENT_DIM_SCALE: f64 = 10.67; const LATENT_DIM_SCALE: f64 = 10.67;
const PRIOR_CIN: usize = 16; const PRIOR_CIN: usize = 16;
@ -354,7 +354,7 @@ fn run(args: Args) -> Result<()> {
)?; )?;
println!("diffusion process with prior {image_embeddings:?}"); 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 = scheduler.timesteps();
let timesteps = &timesteps[..timesteps.len() - 1]; let timesteps = &timesteps[..timesteps.len() - 1];
for (index, &t) in timesteps.iter().enumerate() { for (index, &t) in timesteps.iter().enumerate() {