diff --git a/opensora/models/__init__.py b/opensora/models/__init__.py
index 108f3c2..56db0a5 100644
--- a/opensora/models/__init__.py
+++ b/opensora/models/__init__.py
@@ -4,6 +4,4 @@ from .pixart import *
 from .stdit import *
 from .text_encoder import *
 from .vae import *
-from .vae_3d import *
-from .vae_3d_v2 import *