diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-13 05:05:41 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-13 05:05:41 +0000 |
| commit | 4f73e057a9cb0caaa2adbd7d17ebf221f05821b0 (patch) | |
| tree | a41ad3fb87cf47857f662f10ff898e87c128fab2 /modules/textual_inversion/preprocess.py | |
| parent | 490494320ec8b5e1049c4ff35c3416258b75807b (diff) | |
| parent | 514456101b142b47acf87f6de95bad1a23d73be7 (diff) | |
| download | stable-diffusion-webui-gfx803-4f73e057a9cb0caaa2adbd7d17ebf221f05821b0.tar.gz stable-diffusion-webui-gfx803-4f73e057a9cb0caaa2adbd7d17ebf221f05821b0.tar.bz2 stable-diffusion-webui-gfx803-4f73e057a9cb0caaa2adbd7d17ebf221f05821b0.zip | |
Merge pull request #2324 from HunterVacui/interrogate_include_ranks_in_output
Interrogate: add option to include ranks in output
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)
|