Add the newly added models to the registry

src/skelcast/models/__init__.py

@@ -8,6 +8,8 @@
 from .rnn.lstm import SimpleLSTMRegressor
 from .transformers.transformer import ForecastTransformer
 from .transformers.sttf import SpatioTemporalTransformer
+from .transformers.tflow.tflow import TFlowDiag
+from .transformers.tflow.tfvae import TFlowVAE
 from .rnn.pvred import PositionalVelocityRecurrentEncoderDecoder
 from .rnn.pvred import Encoder, Decoder
 from .cnn.unet import Unet