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
d23a46ce
Commit
d23a46ce
authored
Oct 20, 2022
by
Vladimir Repin
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Different approach to skip/interrupt with highres fix
parent
d1cb08bf
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
7 additions
and
1 deletion
+7
-1
modules/processing.py
modules/processing.py
+3
-1
modules/sd_samplers.py
modules/sd_samplers.py
+4
-0
No files found.
modules/processing.py
View file @
d23a46ce
...
...
@@ -587,7 +587,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
x
=
None
devices
.
torch_gc
()
return
self
.
sampler
.
sample_img2img
(
self
,
samples
,
noise
,
conditioning
,
unconditional_conditioning
,
steps
=
self
.
steps
)
or
samples
samples
=
self
.
sampler
.
sample_img2img
(
self
,
samples
,
noise
,
conditioning
,
unconditional_conditioning
,
steps
=
self
.
steps
)
return
samples
class
StableDiffusionProcessingImg2Img
(
StableDiffusionProcessing
):
...
...
modules/sd_samplers.py
View file @
d23a46ce
...
...
@@ -196,6 +196,7 @@ class VanillaStableDiffusionSampler:
x1
=
self
.
sampler
.
stochastic_encode
(
x
,
torch
.
tensor
([
t_enc
]
*
int
(
x
.
shape
[
0
]))
.
to
(
shared
.
device
),
noise
=
noise
)
self
.
init_latent
=
x
self
.
last_latent
=
x
self
.
step
=
0
samples
=
self
.
launch_sampling
(
steps
,
lambda
:
self
.
sampler
.
decode
(
x1
,
conditioning
,
t_enc
,
unconditional_guidance_scale
=
p
.
cfg_scale
,
unconditional_conditioning
=
unconditional_conditioning
))
...
...
@@ -206,6 +207,7 @@ class VanillaStableDiffusionSampler:
self
.
initialize
(
p
)
self
.
init_latent
=
None
self
.
last_latent
=
x
self
.
step
=
0
steps
=
steps
or
p
.
steps
...
...
@@ -388,6 +390,7 @@ class KDiffusionSampler:
extra_params_kwargs
[
'sigmas'
]
=
sigma_sched
self
.
model_wrap_cfg
.
init_latent
=
x
self
.
last_latent
=
x
samples
=
self
.
launch_sampling
(
steps
,
lambda
:
self
.
func
(
self
.
model_wrap_cfg
,
xi
,
extra_args
=
{
'cond'
:
conditioning
,
'uncond'
:
unconditional_conditioning
,
'cond_scale'
:
p
.
cfg_scale
},
disable
=
False
,
callback
=
self
.
callback_state
,
**
extra_params_kwargs
))
...
...
@@ -414,6 +417,7 @@ class KDiffusionSampler:
else
:
extra_params_kwargs
[
'sigmas'
]
=
sigmas
self
.
last_latent
=
x
samples
=
self
.
launch_sampling
(
steps
,
lambda
:
self
.
func
(
self
.
model_wrap_cfg
,
x
,
extra_args
=
{
'cond'
:
conditioning
,
'uncond'
:
unconditional_conditioning
,
'cond_scale'
:
p
.
cfg_scale
},
disable
=
False
,
callback
=
self
.
callback_state
,
**
extra_params_kwargs
))
return
samples
...
...
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