load_s3gen

Load S3Gen from safetensors file

Description

Load S3Gen from safetensors file

Usage

load_s3gen(path, device = "cpu")

Arguments

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

Value

S3Gen model with loaded weights