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

Print the prior shape.

Browse Source
This commit is contained in:
laurent
2023-09-19 10:26:18 +01:00
parent 5d8e214dfe
commit 3769d8bf71
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -350,7 +350,7 @@ fn run(args: Args) -> Result<()> {
&device, &device,
)?; )?;
println!("diffusion process"); println!("diffusion process with prior {effnet:?}");
let scheduler = wuerstchen::ddpm::DDPMWScheduler::new(60, Default::default())?; let scheduler = wuerstchen::ddpm::DDPMWScheduler::new(60, Default::default())?;
let timesteps = scheduler.timesteps(); let timesteps = scheduler.timesteps();
for (index, &t) in timesteps.iter().enumerate() { for (index, &t) in timesteps.iter().enumerate() {