Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
8d606451
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
12 个月 前同步成功
通知
1771
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
提交
8d606451
编写于
9月 30, 2022
作者:
D
d8ahazard
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix model paths, ensure we have the right files.
Also, clean up logging in the ldsr arch file.
上级
64c6b133
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
8 addition
and
4 deletion
+8
-4
modules/ldsr_model.py
modules/ldsr_model.py
+7
-2
modules/ldsr_model_arch.py
modules/ldsr_model_arch.py
+1
-2
未找到文件。
modules/ldsr_model.py
浏览文件 @
8d606451
...
...
@@ -24,13 +24,18 @@ class UpscalerLDSR(Upscaler):
def
load_model
(
self
,
path
:
str
):
# Remove incorrect project.yaml file if too big
yaml_path
=
os
.
path
.
join
(
self
.
model_path
,
"project.yaml"
)
old_model_path
=
os
.
path
.
join
(
self
.
model_path
,
"model.pth"
)
new_model_path
=
os
.
path
.
join
(
self
.
model_path
,
"model.ckpt"
)
if
os
.
path
.
exists
(
yaml_path
):
statinfo
=
os
.
stat
(
yaml_path
)
if
statinfo
.
st_size
<
=
10485760
:
if
statinfo
.
st_size
>
=
10485760
:
print
(
"Removing invalid LDSR YAML file."
)
os
.
remove
(
yaml_path
)
if
os
.
path
.
exists
(
old_model_path
):
print
(
"Renaming model from model.pth to model.ckpt"
)
os
.
rename
(
old_model_path
,
new_model_path
)
model
=
load_file_from_url
(
url
=
self
.
model_url
,
model_dir
=
self
.
model_path
,
file_name
=
"model.
pth
"
,
progress
=
True
)
file_name
=
"model.
ckpt
"
,
progress
=
True
)
yaml
=
load_file_from_url
(
url
=
self
.
yaml_url
,
model_dir
=
self
.
model_path
,
file_name
=
"project.yaml"
,
progress
=
True
)
...
...
modules/ldsr_model_arch.py
浏览文件 @
8d606451
...
...
@@ -100,7 +100,6 @@ class LDSR:
# If we can adjust the max upscale size, then the 4 below should be our variable
print
(
"Foo"
)
down_sample_rate
=
target_scale
/
4
print
(
f
"Downsample rate is
{
down_sample_rate
}
"
)
wd
=
width_og
*
down_sample_rate
hd
=
height_og
*
down_sample_rate
width_downsampled_pre
=
int
(
wd
)
...
...
@@ -111,7 +110,7 @@ class LDSR:
f
'Downsampling from [
{
width_og
}
,
{
height_og
}
] to [
{
width_downsampled_pre
}
,
{
height_downsampled_pre
}
]'
)
im_og
=
im_og
.
resize
((
width_downsampled_pre
,
height_downsampled_pre
),
Image
.
LANCZOS
)
else
:
print
(
f
"Down sample rate is 1 from
{
target_scale
}
/ 4"
)
print
(
f
"Down sample rate is 1 from
{
target_scale
}
/ 4
(Not downsampling)
"
)
logs
=
self
.
run
(
model
[
"model"
],
im_og
,
diffusion_steps
,
eta
)
sample
=
logs
[
"sample"
]
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录