print_utils.py 328 Bytes
Newer Older
1
2
3
4
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project


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)})")