summaryrefslogtreecommitdiff
path: root/candle-examples/examples/stable-diffusion/utils.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-examples/examples/stable-diffusion/utils.rs')
-rw-r--r--candle-examples/examples/stable-diffusion/utils.rs1
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-examples/examples/stable-diffusion/utils.rs b/candle-examples/examples/stable-diffusion/utils.rs
index 308e577d..5602a9ad 100644
--- a/candle-examples/examples/stable-diffusion/utils.rs
+++ b/candle-examples/examples/stable-diffusion/utils.rs
@@ -1,4 +1,5 @@
use candle::{Device, Result, Tensor};
+use candle_nn::Module;
pub fn linspace(start: f64, stop: f64, steps: usize) -> Result<Tensor> {
if steps < 1 {