(state_dict: Dict[str, Any], old_key: str, new_key: str)
| 185 | |
| 186 | |
| 187 | def update_state_dict_(state_dict: Dict[str, Any], old_key: str, new_key: str) -> dict[str, Any]: |
| 188 | state_dict[new_key] = state_dict.pop(old_key) |
| 189 | |
| 190 | |
| 191 | def rename_transformer_blocks_(key: str, state_dict: Dict[str, Any]): |
no test coverage detected
searching dependent graphs…