Spaces:
Running on Zero
Running on Zero
| def SDTextEncoderStateDictConverter(state_dict): | |
| new_state_dict = {} | |
| for key in state_dict: | |
| if key.startswith("text_model.") and "position_ids" not in key: | |
| new_key = "model." + key | |
| new_state_dict[new_key] = state_dict[key] | |
| return new_state_dict | |