Files
DiffSynth-Studio/diffsynth/utils/state_dict_converters/z_image_text_encoder.py
2026-01-19 16:56:14 +08:00

6 lines
210 B
Python

def ZImageTextEncoderStateDictConverter(state_dict):
state_dict_ = {}
for name in state_dict:
if name != "lm_head.weight":
state_dict_[name] = state_dict[name]
return state_dict_