From ada5d7c096b530fd29b071d798660f3843945e2b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jorge=20Ant=C3=B3nio?= Date: Thu, 28 Mar 2024 09:03:00 +0000 Subject: [PATCH] add send and sync trait bounds for scheduler config in stable diffusion models (#1952) * first commit * add Sync deriving * static * remove static --- candle-transformers/src/models/stable_diffusion/schedulers.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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>; }