Method _regularize_value

tensorrt_llm/parameter.py:265–274

(self, value)

            return None

    def _regularize_value(self, value):
        if isinstance(value, np.ndarray):
            return value

        elif isinstance(value, torch.distributed.tensor.DTensor):
            return value.to_local().cpu().numpy()
        elif isinstance(value, torch.Tensor):
            return torch_to_numpy(value)
        raise TypeError(
            f'Expected numpy.ndarray or torch.Tensor, got {type(value)}')