Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
53924aea
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
11 个月 前同步成功
通知
1758
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
53924aea
编写于
7月 08, 2023
作者:
A
AUTOMATIC1111
提交者:
GitHub
7月 08, 2023
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #11477 from hako-mikan/master
add `before_hr` script callback
上级
953147bf
b0ec69b3
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
17 addition
and
0 deletion
+17
-0
modules/processing.py
modules/processing.py
+3
-0
modules/scripts.py
modules/scripts.py
+14
-0
未找到文件。
modules/processing.py
浏览文件 @
53924aea
...
...
@@ -1078,6 +1078,9 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
sd_models
.
apply_token_merging
(
self
.
sd_model
,
self
.
get_token_merging_ratio
(
for_hr
=
True
))
if
self
.
scripts
is
not
None
:
self
.
scripts
.
before_hr
(
self
)
samples
=
self
.
sampler
.
sample_img2img
(
self
,
samples
,
noise
,
self
.
hr_c
,
self
.
hr_uc
,
steps
=
self
.
hr_second_pass_steps
or
self
.
steps
,
image_conditioning
=
image_conditioning
)
sd_models
.
apply_token_merging
(
self
.
sd_model
,
self
.
get_token_merging_ratio
())
...
...
modules/scripts.py
浏览文件 @
53924aea
...
...
@@ -187,6 +187,11 @@ class Script:
return
f
'script_
{
tabname
}{
title
}
_
{
item_id
}
'
def
before_hr
(
self
,
p
,
*
args
):
"""
This function is called before hires fix start.
"""
pass
current_basedir
=
paths
.
script_path
...
...
@@ -549,6 +554,15 @@ class ScriptRunner:
self
.
scripts
[
si
].
args_to
=
args_to
def
before_hr
(
self
,
p
):
for
script
in
self
.
alwayson_scripts
:
try
:
script_args
=
p
.
script_args
[
script
.
args_from
:
script
.
args_to
]
script
.
before_hr
(
p
,
*
script_args
)
except
Exception
:
errors
.
report
(
f
"Error running before_hr:
{
script
.
filename
}
"
,
exc_info
=
True
)
scripts_txt2img
:
ScriptRunner
=
None
scripts_img2img
:
ScriptRunner
=
None
scripts_postproc
:
scripts_postprocessing
.
ScriptPostprocessingRunner
=
None
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录