diff options
Diffstat (limited to 'candle-examples/examples/stable-diffusion/vae.rs')
| -rw-r--r-- | candle-examples/examples/stable-diffusion/vae.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-examples/examples/stable-diffusion/vae.rs b/candle-examples/examples/stable-diffusion/vae.rs index 7a10d932..abba39fa 100644 --- a/candle-examples/examples/stable-diffusion/vae.rs +++ b/candle-examples/examples/stable-diffusion/vae.rs @@ -10,6 +10,7 @@ use crate::unet_2d_blocks::{ }; use candle::{Result, Tensor}; use candle_nn as nn; +use candle_nn::Module; #[derive(Debug, Clone)] struct EncoderConfig { |