diff options
| -rw-r--r-- | candle-examples/examples/stable-diffusion-3/main.rs | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-examples/examples/stable-diffusion-3/main.rs b/candle-examples/examples/stable-diffusion-3/main.rs index 01b09101..d0bf4bb8 100644 --- a/candle-examples/examples/stable-diffusion-3/main.rs +++ b/candle-examples/examples/stable-diffusion-3/main.rs @@ -180,6 +180,8 @@ fn main() -> Result<()> { let (context, y) = triple.encode_text_to_embedding(prompt.as_str(), &device)?; let (context_uncond, y_uncond) = triple.encode_text_to_embedding(uncond_prompt.as_str(), &device)?; + // Drop the text model early to avoid using too much memory. + drop(triple); let context = Tensor::cat(&[context, context_uncond], 0)?; let y = Tensor::cat(&[y, y_uncond], 0)?; |