| def LTXModelStateDictConverter(state_dict): | |
| state_dict_ = {} | |
| for name in state_dict: | |
| if name.startswith("model.diffusion_model."): | |
| new_name = name.replace("model.diffusion_model.", "") | |
| if new_name.startswith("audio_embeddings_connector.") or new_name.startswith("video_embeddings_connector."): | |
| continue | |
| state_dict_[new_name] = state_dict[name] | |
| return state_dict_ | |