dalle.pt from auxiliary.pt and checkpoints.pt

[robvanvolt]

I disabled wandb during my latest run, and the checkpoints.pt and auxiliary.pt were saved, but unfortunately not the dalle.pt - i am unable to generate images with those pt-files.

Is there a way to generate a dalle.pt from these files (i used deepspeed and taming for training) without having to rerun the whole training procedure?

[janEbert]

Hopefully this script to convert ZeRO to standard checkpoints may help. Other than that, you could get hands on with the new inference API from DeepSpeed. ;)