Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in / Register
Toggle navigation
S
stable-diffusion-webui
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Administrator
stable-diffusion-webui
Commits
9b65c4ec
Unverified
Commit
9b65c4ec
authored
Oct 19, 2022
by
DepFA
Committed by
GitHub
Oct 19, 2022
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
pass preprocess_txt_action param
parent
ab353b14
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
preprocess.py
modules/textual_inversion/preprocess.py
+3
-3
No files found.
modules/textual_inversion/preprocess.py
View file @
9b65c4ec
...
@@ -11,7 +11,7 @@ if cmd_opts.deepdanbooru:
...
@@ -11,7 +11,7 @@ if cmd_opts.deepdanbooru:
import
modules.deepbooru
as
deepbooru
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
,
pr
eprocess_txt_action
,
pr
ocess_flip
,
process_split
,
process_caption
,
process_caption_deepbooru
=
False
):
try
:
try
:
if
process_caption
:
if
process_caption
:
shared
.
interrogator
.
load
()
shared
.
interrogator
.
load
()
...
@@ -21,7 +21,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_
...
@@ -21,7 +21,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_
db_opts
[
deepbooru
.
OPT_INCLUDE_RANKS
]
=
False
db_opts
[
deepbooru
.
OPT_INCLUDE_RANKS
]
=
False
deepbooru
.
create_deepbooru_process
(
opts
.
interrogate_deepbooru_score_threshold
,
db_opts
)
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
,
pr
eprocess_txt_action
,
pr
ocess_flip
,
process_split
,
process_caption
,
process_caption_deepbooru
)
finally
:
finally
:
...
@@ -33,7 +33,7 @@ def preprocess(process_src, process_dst, process_width, process_height, process_
...
@@ -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
,
pr
eprocess_txt_action
,
pr
ocess_flip
,
process_split
,
process_caption
,
process_caption_deepbooru
=
False
):
width
=
process_width
width
=
process_width
height
=
process_height
height
=
process_height
src
=
os
.
path
.
abspath
(
process_src
)
src
=
os
.
path
.
abspath
(
process_src
)
...
...
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