Added preliminary support for SD3.5-large lora training

2024-10-22 12:21:36 -06:00
parent 9f94c7b61e
commit 3400882a80
4 changed files with 176 additions and 16 deletions
--- a/jobs/process/BaseSDTrainProcess.py
+++ b/jobs/process/BaseSDTrainProcess.py
@@ -1907,6 +1907,8 @@ class BaseSDTrainProcess(BaseTrainProcess):
            tags.append("stable-diffusion-xl")
        if self.model_config.is_flux:
            tags.append("flux")
+        if self.model_config.is_v3:
+            tags.append("sd3")
        if self.network_config:
            tags.extend(
                [