load_text_encoder2_weights

Load weights from TorchScript text encoder 2 into native encoder

Description

Load weights from TorchScript text encoder 2 into native encoder

Usage

load_text_encoder2_weights(native_encoder, torchscript_path, verbose = TRUE)

Arguments

  • native_encoder: Native text encoder 2 module
  • torchscript_path: Path to TorchScript encoder .pt file
  • verbose: Print loading progress

Value

The native encoder with loaded weights (invisibly)