diff options
Diffstat (limited to 'candle-examples/examples/stable-diffusion/unet_2d.rs')
| -rw-r--r-- | candle-examples/examples/stable-diffusion/unet_2d.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-examples/examples/stable-diffusion/unet_2d.rs b/candle-examples/examples/stable-diffusion/unet_2d.rs index 0fa2f31a..6f568113 100644 --- a/candle-examples/examples/stable-diffusion/unet_2d.rs +++ b/candle-examples/examples/stable-diffusion/unet_2d.rs @@ -7,6 +7,7 @@ use crate::unet_2d_blocks::*; use crate::utils::{conv2d, Conv2d}; use candle::{Result, Tensor}; use candle_nn as nn; +use candle_nn::Module; #[derive(Debug, Clone, Copy)] pub struct BlockConfig { |