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

[FlaxSpeechEncoderDecoderModel] Skip from_encoder_decoder_pretrained (#16236)

Browse Source 
* skip the test

* fix

* fix skip
This commit is contained in:
Suraj Patil
2022-03-17 20:05:14 +01:00
committed by GitHub
parent b6e06c845f
commit 632ff3c39e
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/speech_encoder_decoder/test_modeling_flax_speech_encoder_decoder.py
View File

@@ -691,6 +691,10 @@ class FlaxWav2Vec2GPT2ModelTest(FlaxEncoderDecoderMixin, unittest.TestCase):
self.assertEqual(len(fx_outputs), len(pt_outputs_loaded), "Output lengths differ between Flax and PyTorch") self.assertEqual(len(fx_outputs), len(pt_outputs_loaded), "Output lengths differ between Flax and PyTorch")
self.assert_almost_equals(fx_logits, pt_logits_loaded.numpy(), 4e-2) self.assert_almost_equals(fx_logits, pt_logits_loaded.numpy(), 4e-2)
@unittest.skip("Re-enable this test once this issue is fixed: https://github.com/google/jax/issues/9941")
def test_encoder_decoder_model_from_encoder_decoder_pretrained(self):
pass
@require_flax @require_flax
class FlaxWav2Vec2BartModelTest(FlaxEncoderDecoderMixin, unittest.TestCase): class FlaxWav2Vec2BartModelTest(FlaxEncoderDecoderMixin, unittest.TestCase):
@@ -807,3 +811,7 @@ class FlaxWav2Vec2BartModelTest(FlaxEncoderDecoderMixin, unittest.TestCase):
self.assertEqual(len(fx_outputs), len(pt_outputs_loaded), "Output lengths differ between Flax and PyTorch") self.assertEqual(len(fx_outputs), len(pt_outputs_loaded), "Output lengths differ between Flax and PyTorch")
self.assert_almost_equals(fx_logits, pt_logits_loaded.numpy(), 4e-2) self.assert_almost_equals(fx_logits, pt_logits_loaded.numpy(), 4e-2)
@unittest.skip("Re-enable this test once this issue is fixed: https://github.com/google/jax/issues/9941")
def test_encoder_decoder_model_from_encoder_decoder_pretrained(self):
pass