huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the clamping for stable-diffusion. (#1041)

Browse Source
This commit is contained in:
Laurent Mazare
2023-10-05 22:20:39 +01:00
committed by GitHub
parent 47c25a567b
commit 716883e9b0
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-examples/examples/stable-diffusion/main.rs
View File

@ -484,9 +484,8 @@ fn run(args: Args) -> Result<()> {
num_samples num_samples
); );
let image = vae.decode(&(&latents / 0.18215)?)?; let image = vae.decode(&(&latents / 0.18215)?)?;
// TODO: Add the clamping between 0 and 1.
let image = ((image / 2.)? + 0.5)?.to_device(&Device::Cpu)?; let image = ((image / 2.)? + 0.5)?.to_device(&Device::Cpu)?;
let image = (image * 255.)?.to_dtype(DType::U8)?.i(0)?; let image = (image.clamp(0f32, 1.)? * 255.)?.to_dtype(DType::U8)?.i(0)?;
let image_filename = output_filename(&final_image, idx + 1, num_samples, None); let image_filename = output_filename(&final_image, idx + 1, num_samples, None);
candle_examples::save_image(&image, image_filename)? candle_examples::save_image(&image, image_filename)?
} }

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

@ -382,8 +382,9 @@ fn run(args: Args) -> Result<()> {
num_samples num_samples
); );
let image = vqgan.decode(&(&latents * 0.3764)?)?; let image = vqgan.decode(&(&latents * 0.3764)?)?;
// TODO: Add the clamping between 0 and 1. let image = (image.clamp(0f32, 1f32)? * 255.)?
let image = (image * 255.)?.to_dtype(DType::U8)?.i(0)?; .to_dtype(DType::U8)?
.i(0)?;
let image_filename = output_filename(&final_image, idx + 1, num_samples, None); let image_filename = output_filename(&final_image, idx + 1, num_samples, None);
candle_examples::save_image(&image, image_filename)? candle_examples::save_image(&image, image_filename)?
} }