From bd6418a940cad7ae88df3d849617f33ad2f5bd9d Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Fri, 1 Aug 2025 21:47:38 +0900
Subject: [PATCH] fix: add assertion for apply_t5_attn_mask requirement in
 Chroma

---
 flux_train_network.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/flux_train_network.py b/flux_train_network.py
index 2d9ab248..cfc61708 100644
--- a/flux_train_network.py
+++ b/flux_train_network.py
@@ -51,6 +51,7 @@ class FluxNetworkTrainer(train_network.NetworkTrainer):
             self.use_clip_l = True
         else:
             self.use_clip_l = False  # Chroma does not use CLIP-L
+            assert args.apply_t5_attn_mask, "apply_t5_attn_mask must be True for Chroma / Chromaではapply_t5_attn_maskを指定する必要があります"
 
         if args.fp8_base_unet:
             args.fp8_base = True  # if fp8_base_unet is enabled, fp8_base is also enabled for FLUX.1