diff --git a/models/turbine_models/tests/stateless_llama_test.py b/models/turbine_models/tests/stateless_llama_test.py index 4b1ffef7..203eef9a 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",