Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
7c3ab416
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,发现更多精彩内容 >>
未验证
提交
7c3ab416
编写于
1月 02, 2024
作者:
A
AUTOMATIC1111
提交者:
GitHub
1月 02, 2024
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #14500 from akx/spandrel-prefer-half
Spandrel: "prefer half" instead of "force half"
上级
51f1cca8
2cacbc12
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
15 addition
and
7 deletion
+15
-7
modules/modelloader.py
modules/modelloader.py
+14
-6
modules/realesrgan_model.py
modules/realesrgan_model.py
+1
-1
未找到文件。
modules/modelloader.py
浏览文件 @
7c3ab416
...
...
@@ -139,23 +139,31 @@ def load_upscalers():
def
load_spandrel_model
(
path
:
str
,
path
:
str
|
os
.
PathLike
,
*
,
device
:
str
|
torch
.
device
|
None
,
half
:
bool
=
False
,
prefer_
half
:
bool
=
False
,
dtype
:
str
|
torch
.
dtype
|
None
=
None
,
expected_architecture
:
str
|
None
=
None
,
)
->
spandrel
.
ModelDescriptor
:
import
spandrel
model_descriptor
=
spandrel
.
ModelLoader
(
device
=
device
).
load_from_file
(
path
)
model_descriptor
=
spandrel
.
ModelLoader
(
device
=
device
).
load_from_file
(
str
(
path
)
)
if
expected_architecture
and
model_descriptor
.
architecture
!=
expected_architecture
:
logger
.
warning
(
f
"Model
{
path
!
r
}
is not a
{
expected_architecture
!
r
}
model (got
{
model_descriptor
.
architecture
!
r
}
)"
,
)
if
half
:
model_descriptor
.
model
.
half
()
half
=
False
if
prefer_half
:
if
model_descriptor
.
supports_half
:
model_descriptor
.
model
.
half
()
half
=
True
else
:
logger
.
info
(
"Model %s does not support half precision, ignoring --half"
,
path
)
if
dtype
:
model_descriptor
.
model
.
to
(
dtype
=
dtype
)
model_descriptor
.
model
.
eval
()
logger
.
debug
(
"Loaded %s from %s (device=%s, half=%s, dtype=%s)"
,
model_descriptor
,
path
,
device
,
half
,
dtype
)
logger
.
debug
(
"Loaded %s from %s (device=%s, half=%s, dtype=%s)"
,
model_descriptor
,
path
,
device
,
half
,
dtype
,
)
return
model_descriptor
modules/realesrgan_model.py
浏览文件 @
7c3ab416
...
...
@@ -39,7 +39,7 @@ class UpscalerRealESRGAN(Upscaler):
model_descriptor
=
modelloader
.
load_spandrel_model
(
info
.
local_data_path
,
device
=
self
.
device
,
half
=
(
not
cmd_opts
.
no_half
and
not
cmd_opts
.
upcast_sampling
),
prefer_
half
=
(
not
cmd_opts
.
no_half
and
not
cmd_opts
.
upcast_sampling
),
expected_architecture
=
"ESRGAN"
,
# "RealESRGAN" isn't a specific thing for Spandrel
)
return
upscale_with_model
(
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录