Merge branch 'merge_punet_sdxl' of https://github.com/nod-ai/SHARK-Tu…

…rbine into merge_punet_sdxl
nod-ai · Aug 13, 2024 · 7ecfece · 7ecfece
2 parents d23a45b + cdd2f66
commit 7ecfece
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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",