mlx.core.load

加载(文件： str, /, 格式： 字符串 | 无 = 无, 返回元数据： 布尔值 = 假, *, 流： None | Stream | Device = 无) → array | dict[str, array]

从二进制文件加载数组。

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

Parameters：

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

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

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

Returns：

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

Return type：

数组 或 字典

警告

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