add void check

huggingface · Sep 2, 2022 · a273710 · a273710
1 parent 5164c9f
commit a273710
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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