File size: 261 Bytes
bc8c4af
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
def WanImageEncoderStateDictConverter(state_dict):
    state_dict_ = {}
    for name in state_dict:
        if name.startswith("textual."):
            continue
        name_ = "model." + name
        state_dict_[name_] = state_dict[name]
    return state_dict_