(state_dict: Dict[str, Any], old_key: str, new_key: str)
| 67 | |
| 68 | |
| 69 | def update_state_dict_(state_dict: Dict[str, Any], old_key: str, new_key: str) -> dict[str, Any]: |
| 70 | state_dict[new_key] = state_dict.pop(old_key) |
| 71 | |
| 72 | |
| 73 | def load_sharded_safetensors(dir: pathlib.Path): |
no test coverage detected
searching dependent graphs…