Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
b0535577
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
11 个月 前同步成功
通知
1771
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
提交
b0535577
编写于
9月 29, 2022
作者:
C
C43H66N12O12S2
提交者:
AUTOMATIC1111
9月 29, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add new samplers
上级
be5555fc
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
10 addition
and
1 deletion
+10
-1
modules/sd_samplers.py
modules/sd_samplers.py
+10
-1
未找到文件。
modules/sd_samplers.py
浏览文件 @
b0535577
...
...
@@ -23,6 +23,8 @@ samplers_k_diffusion = [
(
'Heun'
,
'sample_heun'
,
[
'k_heun'
]),
(
'DPM2'
,
'sample_dpm_2'
,
[
'k_dpm_2'
]),
(
'DPM2 a'
,
'sample_dpm_2_ancestral'
,
[
'k_dpm_2_a'
]),
(
'DPM fast'
,
'sample_dpm_fast'
,
[
'k_dpm_fast'
]),
(
'DPM adaptive'
,
'sample_dpm_adaptive'
,
[
'k_dpm_ad'
]),
]
samplers_data_k_diffusion
=
[
...
...
@@ -37,6 +39,8 @@ samplers = [
SamplerData
(
'PLMS'
,
lambda
model
:
VanillaStableDiffusionSampler
(
ldm
.
models
.
diffusion
.
plms
.
PLMSSampler
,
model
),
[]),
]
samplers_for_img2img
=
[
x
for
x
in
samplers
if
x
.
name
!=
'PLMS'
]
samplers_for_img2img
.
remove
(
samplers_for_img2img
[
6
])
samplers_for_img2img
.
remove
(
samplers_for_img2img
[
6
])
sampler_extra_params
=
{
'sample_euler'
:
[
's_churn'
,
's_tmin'
,
's_tmax'
,
's_noise'
],
...
...
@@ -309,7 +313,12 @@ class KDiffusionSampler:
x
=
x
*
sigmas
[
0
]
extra_params_kwargs
=
self
.
initialize
(
p
)
if
'sigma_min'
in
inspect
.
signature
(
self
.
func
).
parameters
:
if
'n'
in
inspect
.
signature
(
self
.
func
).
parameters
:
samples
=
self
.
func
(
self
.
model_wrap_cfg
,
x
,
sigma_min
=
self
.
model_wrap
.
sigmas
[
0
].
item
(),
sigma_max
=
self
.
model_wrap
.
sigmas
[
-
1
].
item
(),
n
=
steps
,
extra_args
=
{
'cond'
:
conditioning
,
'uncond'
:
unconditional_conditioning
,
'cond_scale'
:
p
.
cfg_scale
},
disable
=
False
,
callback
=
self
.
callback_state
,
**
extra_params_kwargs
)
return
samples
samples
=
self
.
func
(
self
.
model_wrap_cfg
,
x
,
sigma_min
=
self
.
model_wrap
.
sigmas
[
0
].
item
(),
sigma_max
=
self
.
model_wrap
.
sigmas
[
-
1
].
item
(),
extra_args
=
{
'cond'
:
conditioning
,
'uncond'
:
unconditional_conditioning
,
'cond_scale'
:
p
.
cfg_scale
},
disable
=
False
,
callback
=
self
.
callback_state
,
**
extra_params_kwargs
)
return
samples
samples
=
self
.
func
(
self
.
model_wrap_cfg
,
x
,
sigmas
,
extra_args
=
{
'cond'
:
conditioning
,
'uncond'
:
unconditional_conditioning
,
'cond_scale'
:
p
.
cfg_scale
},
disable
=
False
,
callback
=
self
.
callback_state
,
**
extra_params_kwargs
)
return
samples
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录