From b7755ea694c62f0157fe104b55728ce457de3958 Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Thu, 22 Feb 2024 10:25:46 -0800
Subject: [PATCH] improvised karras unet 1d for @MaximilienLC for
 https://github.com/lucidrains/denoising-diffusion-pytorch/issues/295

---
 denoising_diffusion_pytorch/__init__.py | 7 ++++++-
 denoising_diffusion_pytorch/version.py  | 2 +-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/denoising_diffusion_pytorch/__init__.py b/denoising_diffusion_pytorch/__init__.py
index c63fcb09c..9b2699c93 100644
--- a/denoising_diffusion_pytorch/__init__.py
+++ b/denoising_diffusion_pytorch/__init__.py
@@ -8,4 +8,9 @@
 
 from denoising_diffusion_pytorch.denoising_diffusion_pytorch_1d import GaussianDiffusion1D, Unet1D, Trainer1D, Dataset1D
 
-from denoising_diffusion_pytorch.karras_unet import KarrasUnet, InvSqrtDecayLRSched
+from denoising_diffusion_pytorch.karras_unet import (
+    KarrasUnet,
+    InvSqrtDecayLRSched
+)
+
+from denoising_diffusion_pytorch.karras_unet_1d import KarrasUnet1D
\ No newline at end of file
diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
index 461bcf588..919dfc3b9 100644
--- a/denoising_diffusion_pytorch/version.py
+++ b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.10.12'
+__version__ = '1.10.14'