summaryrefslogtreecommitdiff
path: root/candle-examples/examples/stable-diffusion/embeddings.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-examples/examples/stable-diffusion/embeddings.rs')
-rw-r--r--candle-examples/examples/stable-diffusion/embeddings.rs1
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-examples/examples/stable-diffusion/embeddings.rs b/candle-examples/examples/stable-diffusion/embeddings.rs
index c94f24f8..88a153ae 100644
--- a/candle-examples/examples/stable-diffusion/embeddings.rs
+++ b/candle-examples/examples/stable-diffusion/embeddings.rs
@@ -1,6 +1,7 @@
#![allow(dead_code)]
use candle::{Result, Tensor, D};
use candle_nn as nn;
+use candle_nn::Module;
#[derive(Debug)]
pub struct TimestepEmbedding {