Update opensora/models/ae/__init__.py

opensora/models/ae/__init__.py

@@ -1,6 +1,6 @@
-from opensora.models.ae.imagebase import imagebase_ae, imagebase_ae_stride, imagebase_ae_channel
-from opensora.models.ae.videobase import videobase_ae, videobase_ae_stride, videobase_ae_channel
-from opensora.models.ae.videobase import (
+from .imagebase import imagebase_ae, imagebase_ae_stride, imagebase_ae_channel
+from .videobase import videobase_ae, videobase_ae_stride, videobase_ae_channel
+from .videobase import (
     VQVAEConfiguration,
     VQVAEModel,
     VQVAETrainer,
@@ -9,6 +9,7 @@ from opensora.models.ae.videobase import (
     CausalVQVAETrainer
 )
 
+
 ae_stride_config = {}
 ae_stride_config.update(imagebase_ae_stride)
 ae_stride_config.update(videobase_ae_stride)