divamgupta / stable-diffusion-tensorflow

Stable Diffusion in TensorFlow / Keras
Other
1.57k stars 227 forks source link

Is the VAE used in fine-tuning available anywhere? #35

Closed LukeWood closed 1 year ago

LukeWood commented 1 year ago

Hey the @divamgupta ! I'm trying to work on a FineTuning workflow for your model and was wondering if the VAE used in original training is available anywhere in TensorFlow that you are aware of.

I've poked around the repo and did not see any obvious home for it.

If not, no problem - but figured I would ask here first before finding one on my own!