File size: 258 Bytes
bc8c4af
 
 
 
 
 
 
1
2
3
4
5
6
7
def Qwen2ConnectorStateDictConverter(state_dict):
    state_dict_ = {}
    for name in state_dict:
        if name.startswith("connector."):
            name_ = name[len("connector."):]
            state_dict_[name_] = state_dict[name]
    return state_dict_