More generic sampling.

2025-06-19 11:56:45 +00:00 · 2024-09-25 11:15:37 +02:00
parent fa1e0e438e
commit 0bd61bae29
4 changed files with 24 additions and 4 deletions
--- a/candle-transformers/src/models/flux/quantized_model.rs
+++ b/candle-transformers/src/models/flux/quantized_model.rs
@ -577,9 +577,11 @@ impl Flux {
            final_layer,
        })
    }
+}

+impl super::WithForward for Flux {
    #[allow(clippy::too_many_arguments)]
-    pub fn forward(
+    fn forward(
        &self,
        img: &Tensor,
        img_ids: &Tensor,