Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
727e4d10
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
大约 1 年 前同步成功
通知
1784
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
提交
727e4d10
编写于
10月 10, 2022
作者:
A
AUTOMATIC
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
no to different messages plus fix using != to compare to None
上级
b3d3b335
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
3 addition
and
6 deletion
+3
-6
modules/sd_models.py
modules/sd_models.py
+3
-6
未找到文件。
modules/sd_models.py
浏览文件 @
727e4d10
...
...
@@ -152,15 +152,12 @@ def load_model_weights(model, checkpoint_info):
devices
.
dtype_vae
=
torch
.
float32
if
shared
.
cmd_opts
.
no_half
or
shared
.
cmd_opts
.
no_half_vae
else
torch
.
float16
vae_file
=
os
.
path
.
splitext
(
checkpoint_file
)[
0
]
+
".vae.pt"
if
os
.
path
.
exists
(
vae_file
):
print
(
f
"Found VAE Weights:
{
vae_file
}
"
)
elif
shared
.
cmd_opts
.
vae_path
!=
None
:
if
not
os
.
path
.
exists
(
vae_file
)
and
shared
.
cmd_opts
.
vae_path
is
not
None
:
vae_file
=
shared
.
cmd_opts
.
vae_path
print
(
f
'No VAE found for inside the model folder. Using CLI specified :
{
vae_file
}
'
)
else
:
print
(
"No VAE found for inside the model folder. Passing."
)
if
os
.
path
.
exists
(
vae_file
):
print
(
f
"Loading VAE weights from:
{
vae_file
}
"
)
vae_ckpt
=
torch
.
load
(
vae_file
,
map_location
=
"cpu"
)
vae_dict
=
{
k
:
v
for
k
,
v
in
vae_ckpt
[
"state_dict"
].
items
()
if
k
[
0
:
4
]
!=
"loss"
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录