add acestep models

This commit is contained in:
Artiprocher
2026-04-02 10:58:45 +08:00
parent 166e6d2d38
commit 59f512b574
8 changed files with 3169 additions and 1 deletions

View File

@@ -0,0 +1,19 @@
def AceStepTextEncoderStateDictConverter(state_dict):
"""
将 ACE-Step Text Encoder 权重添加 model. 前缀
Args:
state_dict: 原始的 state dict可能是 dict 或 DiskMap
Returns:
转换后的 state dict所有 key 添加 "model." 前缀
"""
state_dict_ = {}
# 处理 DiskMap 或普通 dict
keys = state_dict.keys() if hasattr(state_dict, 'keys') else state_dict
for k in keys:
v = state_dict[k]
if not k.startswith("model."):
k = "model." + k
state_dict_[k] = v
return state_dict_