main
py 10 lines 288 Bytes
Raw
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3 from typing import Any
4
5
6 def tensors_str_no_data(arg: Any):
7 from torch._tensor_str import printoptions
8
9 with printoptions(threshold=1, edgeitems=0):
10 return str(arg)