diff --git a/candle-transformers/src/models/stable_diffusion/schedulers.rs b/candle-transformers/src/models/stable_diffusion/schedulers.rs index 0f0441e0..94f8ab86 100644 --- a/candle-transformers/src/models/stable_diffusion/schedulers.rs +++ b/candle-transformers/src/models/stable_diffusion/schedulers.rs @@ -5,7 +5,7 @@ //! inference speed and quality. use candle::{Result, Tensor}; -pub trait SchedulerConfig: std::fmt::Debug { +pub trait SchedulerConfig: std::fmt::Debug + Send + Sync { fn build(&self, inference_steps: usize) -> Result>; }