Skip to content

Commit d93121a

Browse files
committed
Skip fx output loss bloom model
1 parent 3f88d46 commit d93121a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_modeling_common.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1272,6 +1272,8 @@ def test_torch_fx(self):
12721272
self._create_and_check_torch_fx_tracing(config, inputs_dict)
12731273

12741274
def test_torch_fx_output_loss(self):
1275+
if self.all_model_classes[0].__name__ == "BloomModel":
1276+
self.skipTest(reason="Bloom currently has issues, @michaelbenayoun")
12751277
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
12761278
self._create_and_check_torch_fx_tracing(config, inputs_dict, output_loss=True)
12771279

0 commit comments

Comments
 (0)