Make a CFG version of flux pipeline

2024-08-05 16:35:53 -06:00
parent 99f24cfb0c
commit 272c8608c2
4 changed files with 273 additions and 24 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -370,6 +370,7 @@ class ModelConfig:
        self.is_flux: bool = kwargs.get('is_flux', False)
        if self.is_pixart_sigma:
            self.is_pixart = True
+        self.use_flux_cfg = kwargs.get('use_flux_cfg', False)
        self.is_ssd: bool = kwargs.get('is_ssd', False)
        self.is_vega: bool = kwargs.get('is_vega', False)
        self.is_v_pred: bool = kwargs.get('is_v_pred', False)