VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #157 from shirayu/improve_tag_shuffle

Browse Source 
Always join with ", "
This commit is contained in:
Kohya S
2023-02-07 21:47:05 +09:00
committed by GitHub
parent 5a3d1a57b6 5ea5fefcd2
commit 4fc9f1f8c5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/train_util.py
View File

@@ -265,7 +265,7 @@ class BaseDataset(torch.utils.data.Dataset):
def process_caption(self, caption): def process_caption(self, caption):
if self.shuffle_caption: if self.shuffle_caption:
tokens = caption.strip().split(",") tokens = [t.strip() for t in caption.strip().split(",")]
if self.shuffle_keep_tokens is None: if self.shuffle_keep_tokens is None:
random.shuffle(tokens) random.shuffle(tokens)
else: else:
@@ -274,7 +274,7 @@ class BaseDataset(torch.utils.data.Dataset):
tokens = tokens[self.shuffle_keep_tokens:] tokens = tokens[self.shuffle_keep_tokens:]
random.shuffle(tokens) random.shuffle(tokens)
tokens = keep_tokens + tokens tokens = keep_tokens + tokens
caption = ",".join(tokens).strip() caption = ", ".join(tokens)
for str_from, str_to in self.replacements.items(): for str_from, str_to in self.replacements.items():
if str_from == "": if str_from == "":