(state_dict: dict[str, Any], old_key: str, new_key: str)
| 132 | |
| 133 | |
| 134 | def update_state_dict_inplace(state_dict: dict[str, Any], old_key: str, new_key: str) -> None: |
| 135 | state_dict[new_key] = state_dict.pop(old_key) |
| 136 | |
| 137 | |
| 138 | def remove_keys_inplace(key: str, state_dict: dict[str, Any]) -> None: |
no test coverage detected
searching dependent graphs…