Skip to content
GitLab
菜单
为什么选择 GitLab
定价
联系销售
探索
为什么选择 GitLab
定价
联系销售
探索
登录
获取免费试用
主导航
搜索或转到…
项目
C
ComfyUI
管理
动态
成员
代码
仓库
分支
提交
标签
仓库图
比较修订版本
锁定的文件
部署
模型注册表
分析
模型实验
帮助
帮助
支持
GitLab 文档
比较 GitLab 各版本
社区论坛
为极狐GitLab 提交贡献
提交反馈
隐私声明
快捷键
?
新增功能
4
代码片段
群组
项目
显示更多面包屑
hanamizuki
ComfyUI
提交
25b51b1a
提交
25b51b1a
编辑于
7 months ago
作者:
comfyanonymous
浏览文件
操作
下载
补丁
差异文件
Hunyuan DiT lora support.
上级
61a2b00b
No related branches found
No related tags found
无相关合并请求
变更
1
隐藏空白变更内容
行内
左右并排
显示
1 个更改的文件
comfy/lora.py
+6
-0
6 个添加, 0 个删除
comfy/lora.py
有
6 个添加
和
0 个删除
comfy/lora.py
+
6
−
0
浏览文件 @
25b51b1a
...
...
@@ -282,4 +282,10 @@ def model_lora_keys_unet(model, key_map={}):
key_lora
=
"
transformer.{}
"
.
format
(
k
[:
-
len
(
"
.weight
"
)])
#simpletrainer and probably regular diffusers lora format
key_map
[
key_lora
]
=
to
if
isinstance
(
model
,
comfy
.
model_base
.
HunyuanDiT
):
for
k
in
sdk
:
if
k
.
startswith
(
"
diffusion_model.
"
)
and
k
.
endswith
(
"
.weight
"
):
key_lora
=
k
[
len
(
"
diffusion_model.
"
):
-
len
(
"
.weight
"
)]
key_map
[
"
base_model.model.{}
"
.
format
(
key_lora
)]
=
k
#official hunyuan lora format
return
key_map
此差异已折叠。
点击以展开。
预览
0%
加载中
请重试
或
添加新附件
.
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
保存评论
取消
想要评论请
注册
或
登录