diff options
Diffstat (limited to 'candle-transformers/src/models/flux/sampling.rs')
| -rw-r--r-- | candle-transformers/src/models/flux/sampling.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-transformers/src/models/flux/sampling.rs b/candle-transformers/src/models/flux/sampling.rs index 89b9a953..f3f0eafd 100644 --- a/candle-transformers/src/models/flux/sampling.rs +++ b/candle-transformers/src/models/flux/sampling.rs @@ -92,8 +92,8 @@ pub fn unpack(xs: &Tensor, height: usize, width: usize) -> Result<Tensor> { } #[allow(clippy::too_many_arguments)] -pub fn denoise( - model: &super::model::Flux, +pub fn denoise<M: super::WithForward>( + model: &M, img: &Tensor, img_ids: &Tensor, txt: &Tensor, |