Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
1eb588cb
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
11 个月 前同步成功
通知
1767
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
提交
1eb588cb
编写于
10月 04, 2022
作者:
A
AUTOMATIC
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
remove functools.cache as some people are having issues with it
上级
e1b128d8
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
6 addition
and
2 deletion
+6
-2
modules/prompt_parser.py
modules/prompt_parser.py
+6
-2
未找到文件。
modules/prompt_parser.py
浏览文件 @
1eb588cb
...
...
@@ -29,6 +29,7 @@ def get_learned_conditioning_prompt_schedules(prompts, steps):
%import common.SIGNED_NUMBER -> NUMBER
"""
parser
=
Lark
(
grammar
,
parser
=
'lalr'
)
def
collect_steps
(
steps
,
tree
):
l
=
[
steps
]
class
CollectSteps
(
Visitor
):
...
...
@@ -40,6 +41,7 @@ def get_learned_conditioning_prompt_schedules(prompts, steps):
l
.
append
(
tree
.
children
[
-
1
])
CollectSteps
().
visit
(
tree
)
return
sorted
(
set
(
l
))
def
at_step
(
step
,
tree
):
class
AtStep
(
Transformer
):
def
scheduled
(
self
,
args
):
...
...
@@ -62,11 +64,13 @@ def get_learned_conditioning_prompt_schedules(prompts, steps):
for
child
in
children
:
yield
from
child
return
AtStep
().
transform
(
tree
)
@
functools
.
cache
def
get_schedule
(
prompt
):
tree
=
parser
.
parse
(
prompt
)
return
[[
t
,
at_step
(
t
,
tree
)]
for
t
in
collect_steps
(
steps
,
tree
)]
return
[
get_schedule
(
prompt
)
for
prompt
in
prompts
]
promptdict
=
{
prompt
:
get_schedule
(
prompt
)
for
prompt
in
set
(
prompts
)}
return
[
promptdict
[
prompt
]
for
prompt
in
prompts
]
ScheduledPromptConditioning
=
namedtuple
(
"ScheduledPromptConditioning"
,
[
"end_at_step"
,
"cond"
])
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录