Skip to content

Commit 9e72eb4

Browse files
committed
Skip disk offload test for T5
1 parent b118730 commit 9e72eb4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/models/t5/test_modeling_t5.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -662,6 +662,10 @@ def test_generate_with_head_masking(self):
662662
attn_weights = out[attn_name] if attn_name == attention_names[0] else out[attn_name][-1]
663663
self.assertEqual(sum([w.sum().item() for w in attn_weights]), 0.0)
664664

665+
@unittest.skip("Does not work on the tiny model as we keep hitting edge cases.")
666+
def test_disk_offload(self):
667+
pass
668+
665669

666670
class T5EncoderOnlyModelTester:
667671
def __init__(

0 commit comments

Comments
 (0)