change option position to Training setting

1 files changed, 2 insertions, 3 deletions

diff --git a/modules/textual_inversion/dataset.py b/modules/textual_inversion/dataset.py
index e9d97cc1..df278dc2 100644
--- a/modules/textual_inversion/dataset.py
+++ b/modules/textual_inversion/dataset.py
@@ -24,7 +24,7 @@ class DatasetEntry:
 

 

 class PersonalizedBase(Dataset):

-    def __init__(self, data_root, width, height, repeats, flip_p=0.5, placeholder_token="*", shuffle_tags=True, model=None, device=None, template_file=None, include_cond=False, batch_size=1):

+    def __init__(self, data_root, width, height, repeats, flip_p=0.5, placeholder_token="*", model=None, device=None, template_file=None, include_cond=False, batch_size=1):

         re_word = re.compile(shared.opts.dataset_filename_word_regex) if len(shared.opts.dataset_filename_word_regex) > 0 else None

 

         self.placeholder_token = placeholder_token

@@ -33,7 +33,6 @@ class PersonalizedBase(Dataset):
         self.width = width

         self.height = height

         self.flip = transforms.RandomHorizontalFlip(p=flip_p)

-        self.shuffle_tags = shuffle_tags

 

         self.dataset = []

 

@@ -99,7 +98,7 @@ class PersonalizedBase(Dataset):
     def create_text(self, filename_text):

         text = random.choice(self.lines)

         text = text.replace("[name]", self.placeholder_token)

-        if self.tag_shuffle:

+        if shared.opts.shuffle_tags:

             tags = filename_text.split(',')

             random.shuffle(tags)

             text = text.replace("[filewords]", ','.join(tags))