From a5d5f30fbd9c6c7c78834072401932c84bddaf14 Mon Sep 17 00:00:00 2001 From: gdamms Date: Thu, 5 Feb 2026 15:33:20 +0100 Subject: trying to improve whole project --- models/__init__.py | 6 ++++++ 1 file changed, 6 insertions(+) create mode 100644 models/__init__.py (limited to 'models/__init__.py') diff --git a/models/__init__.py b/models/__init__.py new file mode 100644 index 0000000..db05aa9 --- /dev/null +++ b/models/__init__.py @@ -0,0 +1,6 @@ +""" +Model definitions for MNIST Diffusion. +""" + +from .unet import UNetMNIST +from .autoencoder import AEModule, Autoencoder -- cgit v1.3.1