proper ID

Xilinx · Nov 13, 2024 · ff6ea28 · ff6ea28
1 parent 17328aa
commit ff6ea28
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/brevitas_examples/test_llm.py b/tests/brevitas_examples/test_llm.py
@@ -281,7 +281,8 @@ def test_small_models_acc_pt_ge_2_4(caplog, acc_args_and_acc_pt_ge_2_4):
         "llama-mxfp8",
         "llama-int8-act_equalization=layerwise",
         "mistral-int8-quant-last-layer",
-        "llama-rotation-fx"],
+        "llama-rotation-mixed-fx",
+        "llama-rotation-full-fx",],
     params=[
         {
             "model": "hf-internal-testing/tiny-random-MistralForCausalLM",