From 038dc55a584c1688b77fae7bff2c9515c406efc3 Mon Sep 17 00:00:00 2001 From: "[[ -z $EMAIL ]] && read -e -p \"Enter your email (for git configuration): \" EMAIL" Date: Thu, 6 Feb 2025 10:45:34 -0500 Subject: [PATCH] Skip fx output loss bloom model --- tests/test_modeling_common.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 6d3df2757c2..964595a0cd0 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -1272,6 +1272,8 @@ def test_torch_fx(self): self._create_and_check_torch_fx_tracing(config, inputs_dict) def test_torch_fx_output_loss(self): + if self.all_model_classes[0].__name__ == "BloomModel": + self.skipTest(reason="Bloom currently has issues, @michaelbenayoun") config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() self._create_and_check_torch_fx_tracing(config, inputs_dict, output_loss=True)