Skip to content

Commit

Permalink
skip low_cpu_mem_usage tests (#30782)
Browse files Browse the repository at this point in the history
  • Loading branch information
SunMarc authored May 13, 2024
1 parent 0f8fefd commit 539ed75
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/test_modeling_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
import random
import re
import tempfile
import unittest
import warnings
from collections import defaultdict
from typing import Dict, List, Tuple
Expand Down Expand Up @@ -440,6 +441,7 @@ class CopyClass(model_class):
@slow
@require_accelerate
@mark.accelerate_tests
@unittest.skip("Need to fix since we have a device mismatch")
def test_save_load_low_cpu_mem_usage(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
with tempfile.TemporaryDirectory() as saved_model_path:
Expand All @@ -452,6 +454,7 @@ def test_save_load_low_cpu_mem_usage(self):
@slow
@require_accelerate
@mark.accelerate_tests
@unittest.skip("Need to fix since we have a device mismatch")
def test_save_load_low_cpu_mem_usage_checkpoints(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
with tempfile.TemporaryDirectory() as saved_model_path:
Expand All @@ -465,6 +468,7 @@ def test_save_load_low_cpu_mem_usage_checkpoints(self):
@slow
@require_accelerate
@mark.accelerate_tests
@unittest.skip("Need to fix since we have a device mismatch")
def test_save_load_low_cpu_mem_usage_no_safetensors(self):
with tempfile.TemporaryDirectory() as saved_model_path:
for model_class in self.all_model_classes:
Expand Down

0 comments on commit 539ed75

Please sign in to comment.