Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
S
Stable Diffusion Webui
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
novelai-storage
Stable Diffusion Webui
Commits
f776254b
Commit
f776254b
authored
Oct 12, 2022
by
Greg Fuller
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[2/?] [wip] ignore OPT_INCLUDE_RANKS for training filenames
parent
efefa486
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
2 deletions
+5
-2
modules/deepbooru.py
modules/deepbooru.py
+2
-1
modules/textual_inversion/preprocess.py
modules/textual_inversion/preprocess.py
+3
-1
No files found.
modules/deepbooru.py
View file @
f776254b
...
...
@@ -19,6 +19,7 @@ def get_deepbooru_tags(pil_image):
release_process
()
OPT_INCLUDE_RANKS
=
"include_ranks"
def
create_deepbooru_opts
():
from
modules
import
shared
...
...
@@ -26,7 +27,7 @@ def create_deepbooru_opts():
"use_spaces"
:
shared
.
opts
.
deepbooru_use_spaces
,
"use_escape"
:
shared
.
opts
.
deepbooru_escape
,
"alpha_sort"
:
shared
.
opts
.
deepbooru_sort_alpha
,
"include_ranks"
:
shared
.
opts
.
interrogate_return_ranks
,
OPT_INCLUDE_RANKS
:
shared
.
opts
.
interrogate_return_ranks
,
}
...
...
modules/textual_inversion/preprocess.py
View file @
f776254b
...
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment