From 5f889332882fa2f6db26126075dc4f6e74e685b6 Mon Sep 17 00:00:00 2001
From: hlky <hlky@hlky.ac>
Date: Wed, 18 Dec 2024 11:53:10 +0000
Subject: [PATCH] -skip

---
 tests/models/unets/test_models_unet_2d.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/tests/models/unets/test_models_unet_2d.py b/tests/models/unets/test_models_unet_2d.py
index 59577a54bb22..ddf5f53511f7 100644
--- 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)