how to load safetensors file into the model?

cloneofsimo / minDiffusion

Self-contained, minimalistic implementation of diffusion models with Pytorch.

866 stars 122 forks source link

how to load safetensors file into the model? #7

Open LexiestLeszek opened 1 year ago

LexiestLeszek commented 1 year ago

Hi! If I have a *.safetensors file, can I load it into minDiffusion? I mean can i use either checkpoint or lora downloaded from internet along with minDiffusion?