diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-19 23:49:23 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-19 23:49:23 +0000 |
| commit | 9b65c4ecf4f8eb6187ee721918adebe68e9bc631 (patch) | |
| tree | 4e7dbcf1cb6bd7c3058771d443c274df692b55ad /modules/textual_inversion/preprocess.py | |
| parent | ab353b141df8eee042b0964bcb645015dabf3459 (diff) | |
| download | stable-diffusion-webui-gfx803-9b65c4ecf4f8eb6187ee721918adebe68e9bc631.tar.gz stable-diffusion-webui-gfx803-9b65c4ecf4f8eb6187ee721918adebe68e9bc631.tar.bz2 stable-diffusion-webui-gfx803-9b65c4ecf4f8eb6187ee721918adebe68e9bc631.zip | |
pass preprocess_txt_action param
Diffstat (limited to 'modules/textual_inversion/preprocess.py')
| -rw-r--r-- | modules/textual_inversion/preprocess.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/modules/textual_inversion/preprocess.py b/modules/textual_inversion/preprocess.py index 5c43fe13..3713bc89 100644 --- a/modules/textual_inversion/preprocess.py +++ b/modules/textual_inversion/preprocess.py @@ -11,7 +11,7 @@ if cmd_opts.deepdanbooru: import modules.deepbooru as deepbooru
-def preprocess(process_src, process_dst, process_width, process_height, process_flip, process_split, process_caption, process_caption_deepbooru=False):
+def preprocess(process_src, process_dst, process_width, process_height, preprocess_txt_action, process_flip, process_split, process_caption, process_caption_deepbooru=False):
try:
if process_caption:
shared.interrogator.load()
@@ -21,7 +21,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_ db_opts[deepbooru.OPT_INCLUDE_RANKS] = False
deepbooru.create_deepbooru_process(opts.interrogate_deepbooru_score_threshold, db_opts)
- preprocess_work(process_src, process_dst, process_width, process_height, process_flip, process_split, process_caption, process_caption_deepbooru)
+ preprocess_work(process_src, process_dst, process_width, process_height, preprocess_txt_action, process_flip, process_split, process_caption, process_caption_deepbooru)
finally:
@@ -33,7 +33,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_ -def preprocess_work(process_src, process_dst, process_width, process_height, process_flip, process_split, process_caption, process_caption_deepbooru=False):
+def preprocess_work(process_src, process_dst, process_width, process_height, preprocess_txt_action, process_flip, process_split, process_caption, process_caption_deepbooru=False):
width = process_width
height = process_height
src = os.path.abspath(process_src)
|