Load state dict from path or paths (for sharded model storage) and apply sd_ops
(self, path: str | list[str], sd_ops: SDOps | None = None, device: torch.device | None = None)
| 172 | """Load metadata from path""" |
| 173 | |
| 174 | def load(self, path: str | list[str], sd_ops: SDOps | None = None, device: torch.device | None = None) -> StateDict: |
| 175 | """Load state dict from path or paths (for sharded model storage) and apply sd_ops""" |
| 176 | |
| 177 | |
| 178 | class SafetensorsStateDictLoader(StateDictLoader): |
no outgoing calls