Skip to content

vllm.utils.serial_utils

Functions:

  • numpy2base64 –

    Encode a NumPy array using its .npy representation.

numpy2base64(array)

Encode a NumPy array using its .npy representation.

Source code in vllm/utils/serial_utils.py 
67
68
69
70
71
def numpy2base64(array: np.ndarray) -> str:
    """Encode a NumPy array using its `.npy` representation."""
    with io.BytesIO() as buffer:
        np.save(buffer, array, allow_pickle=False)
        return pybase64.b64encode(buffer.getbuffer()).decode("ascii")