2 files changed, 2 insertions, 6 deletions
diff --git a/candle-examples/examples/stable-diffusion/clip.rs b/candle-examples/examples/stable-diffusion/clip.rs
index be798ad0..227660b1 100644
--- a/candle-examples/examples/stable-diffusion/clip.rs
+++ b/candle-examples/examples/stable-diffusion/clip.rs
@@ -6,6 +6,7 @@
 //!
 //! https://github.com/openai/CLIP
 use candle::{Device, Result, Tensor, D};
+use candle_nn as nn;
 
 #[derive(Debug, Clone, Copy)]
 pub enum Activation {
@@ -16,7 +17,7 @@ pub enum Activation {
 impl Activation {
     fn forward(&self, xs: &Tensor) -> Result<Tensor> {
         match self {
-            Activation::QuickGelu => xs * crate::utils::sigmoid(&(xs * 1.702f64)?)?,
+            Activation::QuickGelu => xs * nn::ops::sigmoid(&(xs * 1.702f64)?)?,
             Activation::Gelu => xs.gelu(),
         }
     }
diff --git a/candle-examples/examples/stable-diffusion/utils.rs b/candle-examples/examples/stable-diffusion/utils.rs
index 90fe3f9a..4294d823 100644
--- a/candle-examples/examples/stable-diffusion/utils.rs
+++ b/candle-examples/examples/stable-diffusion/utils.rs
@@ -1,10 +1,5 @@
 use candle::{Device, Result, Tensor};
 
-pub fn sigmoid(xs: &Tensor) -> Result<Tensor> {
-    // TODO: Add sigmoid as binary ops.
-    (xs.neg()?.exp()? - 1.0)?.recip()
-}
-
 pub fn avg_pool2d(_: &Tensor) -> Result<Tensor> {
     todo!()
 }