add void check

sidthekidder · sidthekidder · commit a273710469fe · 2022-09-02T02:19:51.000-07:00
diff --git a/src/diffusers/models/unet_2d.py b/src/diffusers/models/unet_2d.py
@@ -30,7 +30,7 @@ def __init__(
         attention_head_dim=8,
         norm_num_groups=32,
         norm_eps=1e-5,
-    ):
+    ) -> None:
         super().__init__()
 
         self.sample_size = sample_size
diff --git a/src/diffusers/models/unet_2d_condition.py b/src/diffusers/models/unet_2d_condition.py
@@ -30,7 +30,7 @@ def __init__(
         norm_eps=1e-5,
         cross_attention_dim=1280,
         attention_head_dim=8,
-    ):
+    ) -> None:
         super().__init__()
 
         self.sample_size = sample_size
