A lot of pixart sigma training tweaks

2024-07-28 11:23:18 -06:00
parent 80aa2dbb80
commit 0bc4d555c7
8 changed files with 118 additions and 29 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -128,6 +128,8 @@ class NetworkConfig:
            if self.lorm_config.do_conv:
                self.conv = 4

+        self.transformer_only = kwargs.get('transformer_only', False)
+

 AdapterTypes = Literal['t2i', 'ip', 'ip+', 'clip', 'ilora', 'photo_maker', 'control_net']