A lot of pixart sigma training tweaks

2026-04-27 01:39:20 +00:00 · 2024-07-28 11:23:18 -06:00
parent 80aa2dbb80
commit 0bc4d555c7
8 changed files with 118 additions and 29 deletions
--- a/jobs/process/BaseSDTrainProcess.py
+++ b/jobs/process/BaseSDTrainProcess.py
@@ -1331,6 +1331,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
                    is_lorm=is_lorm,
                    network_config=self.network_config,
                    network_type=self.network_config.type,
+                    transformer_only=self.network_config.transformer_only,
                    **network_kwargs
                )