diff options
Diffstat (limited to 'candle-examples/examples/wuerstchen/main.rs')
| -rw-r--r-- | candle-examples/examples/wuerstchen/main.rs | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/candle-examples/examples/wuerstchen/main.rs b/candle-examples/examples/wuerstchen/main.rs index 8064f87f..bce68114 100644 --- a/candle-examples/examples/wuerstchen/main.rs +++ b/candle-examples/examples/wuerstchen/main.rs @@ -373,7 +373,6 @@ fn run(args: Args) -> Result<()> { ); let image = vqgan.decode(&(&latents * 0.3764)?)?; // TODO: Add the clamping between 0 and 1. - let image = ((image / 2.)? + 0.5)?.to_device(&Device::Cpu)?; let image = (image * 255.)?.to_dtype(DType::U8)?.i(0)?; let image_filename = output_filename(&final_image, idx + 1, num_samples, None); candle_examples::save_image(&image, image_filename)? |