oops

lucidrains · Feb 13, 2024 · 840d3ff · 840d3ff
1 parent c52689d
commit 840d3ff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/denoising_diffusion_pytorch/karras_unet.py b/denoising_diffusion_pytorch/karras_unet.py
@@ -371,7 +371,7 @@ def __init__(
         self.heads = heads
         hidden_dim = dim_head * heads
 
-        self.pixel_norm = PixelNorm(dim = 1)
+        self.pixel_norm = PixelNorm(dim = -1)
 
         self.attend = Attend(flash = flash)
 

diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.10.8'
+__version__ = '1.10.9'