unet_sdxl_native_from_torchscript

Create native SDXL UNet from TorchScript

Description

Create native SDXL UNet from TorchScript

Usage

unet_sdxl_native_from_torchscript(torchscript_path, verbose = TRUE)

Arguments

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

Value

A native SDXL UNet module with loaded weights