fix import VAEStateDictConverter error (#67)

akaitsuki-ii · web-flow · commit e6ec7038a7d8 · 2025-05-20T15:41:52.000+08:00
diff --git a/diffsynth_engine/models/vae/__init__.py b/diffsynth_engine/models/vae/__init__.py
@@ -1,6 +1,3 @@
-from .vae import VAEDecoder, VAEEncoder, VAE
-__all__ = [
-    'VAEDecoder',
-    'VAEEncoder',
-    'VAE'
-]
+from .vae import VAEStateDictConverter, VAEDecoder, VAEEncoder, VAE
+
+__all__ = ["VAEStateDictConverter", "VAEDecoder", "VAEEncoder", "VAE"]
