baofff / U-ViT

A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Autoencoder

Rowerliu opened this issue · comments

I see that you mentioned that: "
Autoencoder
Download stable-diffusion directory from this link (which contains image autoencoders converted from Stable Diffusion). Put the downloaded directory as assets/stable-diffusion in this codebase. The autoencoders are used in latent diffusion models."

but how can i find this original version from SD or LDM? or you retrained the AE and get the new "autoencoder_kl.pth" and "autoencoder_kl_ema.pth"? It makes me a little confused.

Hopes for you reply