Added ability to do prompt attn masking for flux

2026-03-11 05:29:48 +00:00 · 2024-09-02 17:29:36 -06:00
parent d44d4eb61a
commit e5fadddd45
3 changed files with 9 additions and 7 deletions
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -2001,7 +2001,8 @@ class StableDiffusion:
                prompt,
                truncate=not long_prompts,
                max_length=512,
-                dropout_prob=dropout_prob
+                dropout_prob=dropout_prob,
+                attn_mask=self.model_config.attn_masking
            )
            pe = PromptEmbeds(
                prompt_embeds