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
5667ec4c
Unverified
Commit
5667ec4c
authored
Dec 24, 2022
by
AUTOMATIC1111
Committed by
GitHub
Dec 24, 2022
Browse files
Options
Browse Files
Download
Plain Diff
Merge pull request #5797 from mcmonkey4eva/dpm2-a-fix
Add a workaround patch for DPM2 a issue
parents
0fdf368a
7ba9bc2f
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
0 deletions
+6
-0
sd_samplers.py
modules/sd_samplers.py
+6
-0
No files found.
modules/sd_samplers.py
View file @
5667ec4c
...
...
@@ -454,6 +454,9 @@ class KDiffusionSampler:
else
:
sigmas
=
self
.
model_wrap
.
get_sigmas
(
steps
)
if
self
.
funcname
in
[
'sample_dpm_2_ancestral'
,
'sample_dpm_2'
]:
sigmas
=
torch
.
cat
([
sigmas
[:
-
2
],
sigmas
[
-
1
:]])
sigma_sched
=
sigmas
[
steps
-
t_enc
-
1
:]
xi
=
x
+
noise
*
sigma_sched
[
0
]
...
...
@@ -494,6 +497,9 @@ class KDiffusionSampler:
x
=
x
*
sigmas
[
0
]
if
self
.
funcname
in
[
'sample_dpm_2_ancestral'
,
'sample_dpm_2'
]:
sigmas
=
torch
.
cat
([
sigmas
[:
-
2
],
sigmas
[
-
1
:]])
extra_params_kwargs
=
self
.
initialize
(
p
)
if
'sigma_min'
in
inspect
.
signature
(
self
.
func
)
.
parameters
:
extra_params_kwargs
[
'sigma_min'
]
=
self
.
model_wrap
.
sigmas
[
0
]
.
item
()
...
...
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