From cdd2f66ac0f4ef5edea3bd232d56ac2b16df2803 Mon Sep 17 00:00:00 2001 From: Ean Garvey <87458719+monorimet@users.noreply.github.com> Date: Fri, 9 Aug 2024 12:47:35 -0500 Subject: [PATCH] xfail stateless llama test --- models/turbine_models/tests/stateless_llama_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/models/turbine_models/tests/stateless_llama_test.py b/models/turbine_models/tests/stateless_llama_test.py index 4b1ffef73..203eef9a5 100644 --- a/models/turbine_models/tests/stateless_llama_test.py +++ b/models/turbine_models/tests/stateless_llama_test.py @@ -196,6 +196,7 @@ def test_streaming_vmfb_comparison(self): # See: https://github.com/nod-ai/SHARK-Turbine/issues/560 # Developed issues related to the pytorch 2.3 upgrade. + @unittest.expectedFailure def test_rerotated_torch_comparison(self): torch_str = llm_runner.run_torch_llm( "Trelis/Llama-2-7b-chat-hf-function-calling-v2",