feat: adding VQGanVAETaming

ZeroCool940711 · Mar 4, 2023 · ea0e421 · ea0e421
1 parent 9a6333e
commit ea0e421
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,