From 9e72eb441677acfacc5100106c4e2fb9b0b86aa2 Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <Sylvain.gugger@gmail.com>
Date: Tue, 7 Jun 2022 11:11:40 -0400
Subject: [PATCH] Skip disk offload test for T5

---
 tests/models/t5/test_modeling_t5.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/models/t5/test_modeling_t5.py b/tests/models/t5/test_modeling_t5.py
index 1fc1a0a766..3ed5521a62 100644
--- a/tests/models/t5/test_modeling_t5.py
+++ b/tests/models/t5/test_modeling_t5.py
@@ -662,6 +662,10 @@ class T5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
             attn_weights = out[attn_name] if attn_name == attention_names[0] else out[attn_name][-1]
             self.assertEqual(sum([w.sum().item() for w in attn_weights]), 0.0)
 
+    @unittest.skip("Does not work on the tiny model as we keep hitting edge cases.")
+    def test_disk_offload(self):
+        pass
+
 
 class T5EncoderOnlyModelTester:
     def __init__(