improvised karras unet 1d for @MaximilienLC for #295

lucidrains · Feb 22, 2024 · b7755ea · b7755ea
1 parent 505587d
commit b7755ea
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git 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
diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.10.12'
+__version__ = '1.10.14'