Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
184e6701
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,发现更多精彩内容 >>
提交
184e6701
编写于
1月 04, 2023
作者:
A
AUTOMATIC
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix the merge
上级
8839b372
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
5 addition
and
9 deletion
+5
-9
modules/textual_inversion/textual_inversion.py
modules/textual_inversion/textual_inversion.py
+5
-9
未找到文件。
modules/textual_inversion/textual_inversion.py
浏览文件 @
184e6701
...
...
@@ -251,6 +251,7 @@ def validate_train_inputs(model_name, learn_rate, batch_size, gradient_step, dat
if
save_model_every
or
create_image_every
:
assert
log_directory
,
"Log directory is empty"
def
create_dummy_mask
(
x
,
width
=
None
,
height
=
None
):
if
shared
.
sd_model
.
model
.
conditioning_key
in
{
'hybrid'
,
'concat'
}:
...
...
@@ -380,17 +381,12 @@ def train_embedding(embedding_name, learn_rate, batch_size, gradient_step, data_
break
with
devices
.
autocast
():
# c = stack_conds(batch.cond).to(devices.device)
# mask = torch.tensor(batch.emb_index).to(devices.device, non_blocking=pin_memory)
# print(mask)
# c[:, 1:1+embedding.vec.shape[0]] = embedding.vec.to(devices.device, non_blocking=pin_memory)
if
img_c
is
None
:
img_c
=
create_dummy_mask
(
c
,
training_width
,
training_height
)
x
=
batch
.
latent_sample
.
to
(
devices
.
device
,
non_blocking
=
pin_memory
)
c
=
shared
.
sd_model
.
cond_stage_model
(
batch
.
cond_text
)
if
img_c
is
None
:
img_c
=
create_dummy_mask
(
c
,
training_width
,
training_height
)
cond
=
{
"c_concat"
:
[
img_c
],
"c_crossattn"
:
[
c
]}
loss
=
shared
.
sd_model
(
x
,
cond
)[
0
]
/
gradient_step
del
x
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录