14 lines
504 B
Python
14 lines
504 B
Python
from .vae.vae import HFVAEWrapper
|
|
from .vae.vae import SDVAEWrapper
|
|
from .vqvae.vqvae import SDVQVAEWrapper
|
|
|
|
vae = ['stabilityai/sd-vae-ft-mse', 'stabilityai/sd-vae-ft-ema']
|
|
vqvae = ['vqgan_imagenet_f16_1024', 'vqgan_imagenet_f16_16384', 'vqgan_gumbel_f8']
|
|
|
|
imagebase_ae = {
|
|
'stabilityai/sd-vae-ft-mse': HFVAEWrapper,
|
|
'stabilityai/sd-vae-ft-ema': HFVAEWrapper,
|
|
'vqgan_imagenet_f16_1024': SDVQVAEWrapper,
|
|
'vqgan_imagenet_f16_16384': SDVQVAEWrapper,
|
|
'vqgan_gumbel_f8': SDVQVAEWrapper,
|
|
} |