Update all files for EO-VAE
Browse files- _eo_vae/__init__.py +5 -0
_eo_vae/__init__.py
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# EO-VAE: self-contained implementation for diffusers-style loading
|
| 2 |
+
from .modeling import EOVAEModel, Encoder, Decoder
|
| 3 |
+
from .distributions import DiagonalGaussianDistribution
|
| 4 |
+
|
| 5 |
+
__all__ = ["EOVAEModel", "Encoder", "Decoder", "DiagonalGaussianDistribution"]
|