debug

model.py

@@ -148,7 +148,7 @@ class UNet(nn.Module):
                     out_channels: int,
                     block_channels: int = None,
                     return_list=False
-                    ) -> list | nn.Sequential:
+                    ) -> list:
         if not block_channels:
             block_channels = in_channels // 2
         block = []
@@ -171,7 +171,7 @@ class UNet(nn.Module):
                   out_channels: int,
                   return_list=False,
                   max_pool=True,
-                  ) -> list | nn.Sequential:
+                  ) -> list:
         if max_pool:
             block = [nn.MaxPool2d(kernel_size=self.max_pool_kernel_size)]
         else: