| 647 | |
| 648 | |
| 649 | class _StateDictModule(torch.nn.Module): |
| 650 | def __init__(self, state_dict: Mapping[str, torch.Tensor]): |
| 651 | super().__init__() |
| 652 | self._state_dict = dict(state_dict) |
| 653 | |
| 654 | def state_dict(self, *args: Any, **kwargs: Any) -> Mapping[str, torch.Tensor]: |
| 655 | del args, kwargs |
| 656 | return self._state_dict |
| 657 | |
| 658 | |
| 659 | def _run_vllm_cuda_vmm_external_storage(args: argparse.Namespace) -> dict[str, Any]: |
no outgoing calls
no test coverage detected