divamgupta / stable-diffusion-tensorflow

Stable Diffusion in TensorFlow / Keras
Other
1.58k stars 228 forks source link

Script to convert PyTorch weights #23

Open jorgemcgomes opened 2 years ago

jorgemcgomes commented 2 years ago

Thanks and congratz for this awesome piece of work.

Could you add to the repo whatever script was used to convert the original SD PyTorch checkpoint to the TF weights used by this repo?

That would allow people to load fine-tuned pytorch models into tensorflow for inference purposes.