File size: 210 Bytes
bc8c4af | 1 2 3 4 5 6 | 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_ |