Loading Intermediate Checkpoints for Inference

[mnslarcher]

I'm attempting to perform inference using an intermediate checkpoint, but I believe the code in infer.ipynb might not be up to date. Currently, the code looks like this:

transformer = Transformer2DModel.from_pretrained("/workspace/PixArt-alpha/output/diffusers_trained/transformer", torch_dtype=torch.float16)
pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-512x512", torch_dtype=torch.float16, transformer=transformer)
pipe = pipe.to("cuda")

At this stage, checkpoints are being saved as .pth files in the following structure:

output
  trained_models
    checkpoints
      epoch_1_step_100.pth
      ...
      epoch_4_step_700.pth

Could anyone provide suggestions on how to correctly load an intermediate checkpoint for inference?

[mnslarcher]

Apologies, I forgot that first one have to use this:

convert-checkpoint-to-diffusers.ipynb

I'll try again!