fixed bug with trigger word embedding. Allow control images to load from the dataloader or legacy way

2026-03-05 02:29:49 +00:00 · 2023-10-09 06:21:49 -06:00
parent b1cfafa0c6
commit 1d3de678aa
2 changed files with 6 additions and 3 deletions
--- a/jobs/process/BaseSDTrainProcess.py
+++ b/jobs/process/BaseSDTrainProcess.py
@@ -231,8 +231,8 @@ class BaseSDTrainProcess(BaseTrainProcess):
        if self.trigger_word is not None:
            # just so auto1111 will pick it up
            o_dict['ss_tag_frequency'] = {
-                [self.trigger_word ]: {
-                    [self.trigger_word ]: 1
+                f"1_{self.trigger_word}": {
+                    f"{self.trigger_word}": 1
                }
            }