From ea0e421793d4e366aa1bd3d2363c1952cb8f049f Mon Sep 17 00:00:00 2001 From: wnakano Date: Sat, 4 Mar 2023 02:11:26 -0300 Subject: [PATCH] feat: adding VQGanVAETaming --- muse_maskgit_pytorch/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/muse_maskgit_pytorch/__init__.py b/muse_maskgit_pytorch/__init__.py index c9169fd..0519d95 100644 --- a/muse_maskgit_pytorch/__init__.py +++ b/muse_maskgit_pytorch/__init__.py @@ -1,4 +1,5 @@ from muse_maskgit_pytorch.vqgan_vae import VQGanVAE +from muse_maskgit_pytorch.vqgan_vae_taming import VQGanVAETaming from muse_maskgit_pytorch.muse_maskgit_pytorch import ( Transformer, MaskGit,