load_unet_sdxl_weights

Load weights from TorchScript SDXL UNet into native SDXL UNet

Description

Load weights from TorchScript SDXL UNet into native SDXL UNet

Usage

load_unet_sdxl_weights(native_unet, torchscript_path, verbose = TRUE)

Arguments

  • native_unet: Native SDXL UNet module
  • torchscript_path: Path to TorchScript SDXL UNet .pt file
  • verbose: Print loading progress

Value

The native UNet with loaded weights (invisibly)