Skip to content

Commit

Permalink
Revert "Patch to skip failing test_save_load_low_cpu_mem_usage tests (
Browse files Browse the repository at this point in the history
huggingface#29043)"

This reverts commit 4156f51.
  • Loading branch information
amyeroberts committed Feb 20, 2024
1 parent 858a23f commit 75e514d
Show file tree
Hide file tree
Showing 6 changed files with 0 additions and 40 deletions.
6 changes: 0 additions & 6 deletions tests/models/bert_generation/test_modeling_bert_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -305,12 +305,6 @@ def test_model_from_pretrained(self):
model = BertGenerationEncoder.from_pretrained("google/bert_for_seq_generation_L-24_bbc_encoder")
self.assertIsNotNone(model)

@unittest.skip(
"Not currently compatible. Fails with - NotImplementedError: Cannot copy out of meta tensor; no data!"
)
def test_save_load_low_cpu_mem_usage(self):
pass


@require_torch
class BertGenerationEncoderIntegrationTest(unittest.TestCase):
Expand Down
6 changes: 0 additions & 6 deletions tests/models/fsmt/test_modeling_fsmt.py
Original file line number Diff line number Diff line change
Expand Up @@ -329,12 +329,6 @@ def test_tie_model_weights(self):
def test_resize_embeddings_untied(self):
pass

@unittest.skip(
"Not currently compatible. Fails with - NotImplementedError: Cannot copy out of meta tensor; no data!"
)
def test_save_load_low_cpu_mem_usage(self):
pass


@require_torch
class FSMTHeadTests(unittest.TestCase):
Expand Down
6 changes: 0 additions & 6 deletions tests/models/marian/test_modeling_marian.py
Original file line number Diff line number Diff line change
Expand Up @@ -372,12 +372,6 @@ def test_training_gradient_checkpointing_use_reentrant(self):
def test_training_gradient_checkpointing_use_reentrant_false(self):
pass

@unittest.skip(
"Not currently compatible. Fails with - NotImplementedError: Cannot copy out of meta tensor; no data!"
)
def test_save_load_low_cpu_mem_usage(self):
pass


def assert_tensors_close(a, b, atol=1e-12, prefix=""):
"""If tensors have different shapes, different values or a and b are not both tensors, raise a nice Assertion error."""
Expand Down
4 changes: 0 additions & 4 deletions tests/models/musicgen/test_modeling_musicgen.py
Original file line number Diff line number Diff line change
Expand Up @@ -1144,10 +1144,6 @@ def test_greedy_generate_stereo_outputs(self):

self.assertNotIn(config.pad_token_id, output_generate)

@unittest.skip("Fails with - TypeError: _weight_norm_interface() missing 1 required positional argument: 'dim'")
def test_save_load_low_cpu_mem_usage(self):
pass


def get_bip_bip(bip_duration=0.125, duration=0.5, sample_rate=32000):
"""Produces a series of 'bip bip' sounds at a given frequency."""
Expand Down
12 changes: 0 additions & 12 deletions tests/models/reformer/test_modeling_reformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -687,12 +687,6 @@ def _check_hidden_states_for_generate(
def test_left_padding_compatibility(self):
pass

@unittest.skip(
"Not currently compatible. Fails with - NotImplementedError: Cannot copy out of meta tensor; no data!"
)
def test_save_load_low_cpu_mem_usage(self):
pass


@require_torch
class ReformerLSHAttnModelTest(
Expand Down Expand Up @@ -854,12 +848,6 @@ def test_past_key_values_format(self):
def test_left_padding_compatibility(self):
pass

@unittest.skip(
"Not currently compatible. Fails with - NotImplementedError: Cannot copy out of meta tensor; no data!"
)
def test_save_load_low_cpu_mem_usage(self):
pass


@require_torch
@require_sentencepiece
Expand Down
6 changes: 0 additions & 6 deletions tests/models/xlm_roberta_xl/test_modeling_xlm_roberta_xl.py
Original file line number Diff line number Diff line change
Expand Up @@ -515,12 +515,6 @@ def test_create_position_ids_from_inputs_embeds(self):
self.assertEqual(position_ids.shape, expected_positions.shape)
self.assertTrue(torch.all(torch.eq(position_ids, expected_positions)))

@unittest.skip(
"Not currently compatible. Fails with - NotImplementedError: Cannot copy out of meta tensor; no data!"
)
def test_save_load_low_cpu_mem_usage(self):
pass


@require_torch
class XLMRobertaModelXLIntegrationTest(unittest.TestCase):
Expand Down

0 comments on commit 75e514d

Please sign in to comment.