read_safetensors

Read safetensors file

Description

Read safetensors file

Usage

read_safetensors(path, device = "cpu")

Arguments

  • path: Path to .safetensors file
  • device: Device to load tensors to (“cpu”, “cuda”, etc.)

Value

Named list of torch tensors