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
72022133
Unverified
Commit
72022133
authored
Feb 19, 2023
by
AUTOMATIC1111
Committed by
GitHub
Feb 19, 2023
Browse files
Options
Browse Files
Download
Plain Diff
Merge pull request #7571 from mezotaken/int_loop
Optionally append interrogated prompt in loopback script
parents
d84f3cf7
7dd23973
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
12 additions
and
2 deletions
+12
-2
loopback.py
scripts/loopback.py
+12
-2
No files found.
scripts/loopback.py
View file @
72022133
...
...
@@ -8,6 +8,7 @@ from modules import processing, shared, sd_samplers, images
from
modules.processing
import
Processed
from
modules.sd_samplers
import
samplers
from
modules.shared
import
opts
,
cmd_opts
,
state
from
modules
import
deepbooru
class
Script
(
scripts
.
Script
):
...
...
@@ -20,10 +21,11 @@ class Script(scripts.Script):
def
ui
(
self
,
is_img2img
):
loops
=
gr
.
Slider
(
minimum
=
1
,
maximum
=
32
,
step
=
1
,
label
=
'Loops'
,
value
=
4
,
elem_id
=
self
.
elem_id
(
"loops"
))
denoising_strength_change_factor
=
gr
.
Slider
(
minimum
=
0.9
,
maximum
=
1.1
,
step
=
0.01
,
label
=
'Denoising strength change factor'
,
value
=
1
,
elem_id
=
self
.
elem_id
(
"denoising_strength_change_factor"
))
append_interrogation
=
gr
.
Dropdown
(
label
=
"Append interrogated prompt at each iteration"
,
choices
=
[
"None"
,
"CLIP"
,
"DeepBooru"
],
value
=
"None"
)
return
[
loops
,
denoising_strength_change_factor
]
return
[
loops
,
denoising_strength_change_factor
,
append_interrogation
]
def
run
(
self
,
p
,
loops
,
denoising_strength_change_factor
):
def
run
(
self
,
p
,
loops
,
denoising_strength_change_factor
,
append_interrogation
):
processing
.
fix_seed
(
p
)
batch_count
=
p
.
n_iter
p
.
extra_generation_params
=
{
...
...
@@ -40,6 +42,7 @@ class Script(scripts.Script):
grids
=
[]
all_images
=
[]
original_init_image
=
p
.
init_images
original_prompt
=
p
.
prompt
state
.
job_count
=
loops
*
batch_count
initial_color_corrections
=
[
processing
.
setup_color_correction
(
p
.
init_images
[
0
])]
...
...
@@ -58,6 +61,13 @@ class Script(scripts.Script):
if
opts
.
img2img_color_correction
:
p
.
color_corrections
=
initial_color_corrections
if
append_interrogation
!=
"None"
:
p
.
prompt
=
original_prompt
+
", "
if
original_prompt
!=
""
else
""
if
append_interrogation
==
"CLIP"
:
p
.
prompt
+=
shared
.
interrogator
.
interrogate
(
p
.
init_images
[
0
])
elif
append_interrogation
==
"DeepBooru"
:
p
.
prompt
+=
deepbooru
.
model
.
tag
(
p
.
init_images
[
0
])
state
.
job
=
f
"Iteration {i + 1}/{loops}, batch {n + 1}/{batch_count}"
processed
=
processing
.
process_images
(
p
)
...
...
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