diffusion_rs_common::core::pickle

Function read_pth_tensor_info

source 
pub fn read_pth_tensor_info<P: AsRef<Path>>(
    file: P,
    verbose: bool,
    key: Option<&str>,
) -> Result<Vec<TensorInfo>>
Expand description

Read the tensor info from a .pth file.

ยงArguments

  • file - The path to the .pth file.
  • verbose - Whether to print debug information.
  • key - Optional key to retrieve state_dict from the pth file.