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

This commit is contained in:
Jaret Burkett committed 2023-10-09 06:21:49 -06:00
1 parent b1cfafa0c6
commit 1d3de678aa
2 files changed
+6 -3

No files matched your search

+2 -2
View File
@@ -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
}
}