load_s3gen_weights

Load S3Gen weights

Description

Load S3Gen weights

Usage

load_s3gen_weights(model, state_dict)

Arguments

  • model: S3Gen model
  • state_dict: State dictionary from safetensors

Value

Model with loaded weights