From 6576e98498559974260ff1a66f09f47b3f460497 Mon Sep 17 00:00:00 2001
From: Tatsuo Okubo <tatsuo.okubo@gmail.com>
Date: Mon, 6 Nov 2023 21:48:39 +0800
Subject: [PATCH] minor typo in the type hint of ddpm/unet

---
 labml_nn/diffusion/ddpm/unet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/labml_nn/diffusion/ddpm/unet.py b/labml_nn/diffusion/ddpm/unet.py
index 4a2a0572..a4b36d09 100644
--- a/labml_nn/diffusion/ddpm/unet.py
+++ b/labml_nn/diffusion/ddpm/unet.py
@@ -312,7 +312,7 @@ class UNet(Module):
 
     def __init__(self, image_channels: int = 3, n_channels: int = 64,
                  ch_mults: Union[Tuple[int, ...], List[int]] = (1, 2, 2, 4),
-                 is_attn: Union[Tuple[bool, ...], List[int]] = (False, False, True, True),
+                 is_attn: Union[Tuple[bool, ...], List[bool]] = (False, False, True, True),
                  n_blocks: int = 2):
         """
         * `image_channels` is the number of channels in the image. $3$ for RGB.