Some work on sd3 training. Not working

2024-06-13 12:19:16 -06:00
parent cb5d28cba9
commit bd10d2d668
12 changed files with 306 additions and 36 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -336,6 +336,7 @@ class ModelConfig:
        self.is_xl: bool = kwargs.get('is_xl', False)
        self.is_pixart: bool = kwargs.get('is_pixart', False)
        self.is_pixart_sigma: bool = kwargs.get('is_pixart', False)
+        self.is_v3: bool = kwargs.get('is_v3', False)
        if self.is_pixart_sigma:
            self.is_pixart = True
        self.is_ssd: bool = kwargs.get('is_ssd', False)