SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix CPU offload + disk offload tests (#27204)

Browse Source 
Fix disk offload tests + weight sharing issues
This commit is contained in:
Lysandre Debut
2023-11-01 19:25:23 +01:00
committed by GitHub
parent c9e72f55b2
commit 95020f208e
14 changed files with 155 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/vitdet/test_modeling_vitdet.py
View File

@@ -182,7 +182,11 @@ class VitDetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
# TODO: Fix me (once this model gets more usage)
@unittest.skip("Does not work on the tiny model as we keep hitting edge cases.")
def test_disk_offload(self):
def test_disk_offload_bin(self):
super().test_disk_offload()
@unittest.skip("Does not work on the tiny model as we keep hitting edge cases.")
def test_disk_offload_safetensors(self):
super().test_disk_offload()
# TODO: Fix me (once this model gets more usage)