Writes a list of tensors to the safetensors format — safe_save_file • safetensors

Writes a list of tensors to the safetensors format

Usage

safe_save_file(tensors, path, ..., metadata = NULL)

safe_serialize(tensors, ..., metadata = NULL)

Arguments

tensors: A named list of tensors. Currently only torch tensors are supported.
path: The path to save the tensors to. It can also be a binary connection, as eg, created with file().
...: Currently unused.
metadata: An optional string that is added to the file header. Possibly adding additional description to the weights.

Value

The path invisibly or a raw vector.

Functions

safe_serialize(): Serializes the tensors and returns a raw vector.

Examples

if (rlang::is_installed("torch") && torch::torch_is_installed()) {
  tensors <- list(x = torch::torch_randn(10, 10))
  temp <- tempfile()
  safe_save_file(tensors, temp)
  safe_load_file(temp)

  ser <- safe_serialize(tensors)
}