base setup

sambhavnoobcoder · Jan 27, 2025 · 077a14c · 077a14c
1 parent 78154a3
commit 077a14c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/models/mask2former/test_initialization_mask2former.py b/tests/models/mask2former/test_initialization_mask2former.py
@@ -5,3 +5,9 @@
     Mask2FormerMaskedAttentionDecoderLayer,
     Mask2FormerPixelDecoderEncoderMultiscaleDeformableAttention
 )
+
+class TestMask2FormerInitialization(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.config = Mask2FormerConfig()
+        cls.model = Mask2FormerModel(cls.config)