fixed bug with trigger word embedding. Allow control images to load from the dataloader or legacy way
This commit is contained in:
1 parent
b1cfafa0c6
commit
1d3de678aa
2 files changed
+6
-3
No files matched your search
@@ -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
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user