diff options
| author | yfszzx <yfszzx@gmail.com> | 2022-10-14 03:51:26 +0000 |
|---|---|---|
| committer | yfszzx <yfszzx@gmail.com> | 2022-10-14 03:51:26 +0000 |
| commit | d48f3470c8a86f22dce106e7d3c1803e5ba92085 (patch) | |
| tree | 2526ceaffa0177b40cecc30854f66a524cf6ac71 /modules/textual_inversion/preprocess.py | |
| parent | 4a37c7eedeab579efec03e8dae3f3f9fd4a37b02 (diff) | |
| parent | 7c8903367c350e7e3ba8f90679890566446f36b2 (diff) | |
| download | stable-diffusion-webui-gfx803-d48f3470c8a86f22dce106e7d3c1803e5ba92085.tar.gz stable-diffusion-webui-gfx803-d48f3470c8a86f22dce106e7d3c1803e5ba92085.tar.bz2 stable-diffusion-webui-gfx803-d48f3470c8a86f22dce106e7d3c1803e5ba92085.zip | |
Merge branch 'master' of https://github.com/yfszzx/stable-diffusion-webui-plus
Diffstat (limited to 'modules/textual_inversion/preprocess.py')
| -rw-r--r-- | modules/textual_inversion/preprocess.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/textual_inversion/preprocess.py b/modules/textual_inversion/preprocess.py index 3047bede..886cf0c3 100644 --- a/modules/textual_inversion/preprocess.py +++ b/modules/textual_inversion/preprocess.py @@ -17,7 +17,9 @@ def preprocess(process_src, process_dst, process_width, process_height, process_ shared.interrogator.load()
if process_caption_deepbooru:
- deepbooru.create_deepbooru_process(opts.interrogate_deepbooru_score_threshold, deepbooru.create_deepbooru_opts())
+ db_opts = deepbooru.create_deepbooru_opts()
+ 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)
|