load_decoder_weights

Load weights from TorchScript decoder into native decoder

Description

Load weights from TorchScript decoder into native decoder

Usage

load_decoder_weights(native_decoder, torchscript_path, verbose = TRUE)

Arguments

  • native_decoder: Native VAE decoder module
  • torchscript_path: Path to TorchScript decoder .pt file
  • verbose: Print loading progress

Value

The native decoder with loaded weights (invisibly)