mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-04-16 07:08:19 +00:00
079e51c9f3
* auto intergrate joyimage model * joyimage pipeline * train * ready * styling * joyai-image docs * update readme * pr review
21 lines
789 B
Python
21 lines
789 B
Python
def JoyAIImageTextEncoderStateDictConverter(state_dict):
|
|
"""Convert HuggingFace Qwen3VL checkpoint keys to DiffSynth wrapper keys.
|
|
|
|
Mapping (checkpoint -> wrapper):
|
|
- lm_head.weight -> model.lm_head.weight
|
|
- model.language_model.* -> model.model.language_model.*
|
|
- model.visual.* -> model.model.visual.*
|
|
"""
|
|
state_dict_ = {}
|
|
for key in state_dict:
|
|
if key == "lm_head.weight":
|
|
new_key = "model.lm_head.weight"
|
|
elif key.startswith("model.language_model."):
|
|
new_key = "model.model." + key[len("model."):]
|
|
elif key.startswith("model.visual."):
|
|
new_key = "model.model." + key[len("model."):]
|
|
else:
|
|
new_key = key
|
|
state_dict_[new_key] = state_dict[key]
|
|
return state_dict_
|