mlx.core.load

目录

mlx.core.load#

load(file: str, /, format: str | None = None, return_metadata: bool = False, *, stream: None | Stream |Device = None) → array | dict[str, array]#

从二进制文件加载一个或多个数组。

支持的格式有 .npy, .npz, .safetensors 和 .gguf。

参数:

  • file (文件, str) – 保存数组的文件。

  • format (str, 可选) – 文件格式。如果为 None，则从文件扩展名推断格式。支持的格式：npy, npz, 和 safetensors。默认值：None。

  • return_metadata (bool, 可选) – 为支持元数据的格式加载元数据。元数据将作为附加字典返回。默认值：False。

返回:

如果从 .npy 文件加载，则返回单个数组；如果从 .npz 或 .safetensors 文件加载，则返回一个将名称映射到数组的字典。如果 return_metadata 为 True，将额外返回一个元数据字典。

返回类型:

数组 或 字典

警告

从 GGUF 加载不支持的量化格式时，张量将自动转换为 mx.float16 类型

目录