您正在查看 主分支 版本,需要从源代码安装。如果您想要使用常规的 pip 安装,请查看最新的稳定版本 (v0.3.2)。
Torch API
safetensors.torch.load_file
< 源代码 >( filename: Union device: Union = 'cpu' ) → Dict[str, torch.Tensor]
将 safetensors 文件加载到 torch 格式。
safetensors.torch.load
< 源代码 >( data: bytes ) → Dict[str, torch.Tensor]
从纯字节中将 safetensors 文件加载到 torch 格式。
safetensors.torch.save_file
< 源代码 >( tensors: Dict filename: Union metadata: Optional = None ) → None
将张量字典保存为 safetensors 格式的原始字节。
safetensors.torch.save
< 源代码 >( 张量: Dict 元数据: Optional = None ) → 字节
将张量字典保存为 safetensors 格式的原始字节。
safetensors.torch.load_model
< 源代码 >( 模型: Module 文件名: Union 严格: bool = True 设备: Union = 'cpu' ) → `(缺失,意外)`
将给定文件名加载到 PyTorch 模型中。此方法专门用于避免 safetensors
中不允许的张量共享问题。 有关张量共享的更多信息
safetensors.torch.save_model
< 源代码 > ( model: Module filename: str metadata: Optional = None force_contiguous: bool = True )
将给定的 PyTorch 模型保存到指定的文件名中。此方法专门用于避免 safetensors
中不允许的张量共享问题。 有关张量共享的更多信息