main
py 7 lines 328 Bytes
Raw
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4
5 def print_embeddings(embeds: list[float], prefix: str = "Embeddings"):
6 embeds_trimmed = (str(embeds[:4])[:-1] + ", ...]") if len(embeds) > 4 else embeds
7 print(f"{prefix}: {embeds_trimmed} (size={len(embeds)})")