-skip

dg845 · Dec 18, 2024 · 5f88933 · 5f88933
1 parent e619ec8
commit 5f88933
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/tests/models/unets/test_models_unet_2d.py b/tests/models/unets/test_models_unet_2d.py
@@ -361,10 +361,6 @@ def test_forward_with_norm_groups(self):
     def test_gradient_checkpointing_is_applied(self):
         expected_set = {
             "UNetMidBlock2D",
-            "SkipDownBlock2D",
-            "AttnSkipDownBlock2D",
-            "SkipUpBlock2D",
-            "AttnSkipUpBlock2D",
         }
 
         block_out_channels = (32, 64, 64, 64)