From 77fbd9e225a740b54f3281d3bbb1dd54b03d99e5 Mon Sep 17 00:00:00 2001
From: Andreas Karatzas <akaratza@amd.com>
Date: Sun, 27 Sep 2026 22:47:41 -0500
Subject: [PATCH] [Test][ROCm] Stabilize the mixed OLMoE LoRA test (#58945)

Signed-off-by: Andreas Karatzas <andreas.karatzas@protonmail.com>
---
 tests/lora/test_olmoe_tp.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/lora/test_olmoe_tp.py b/tests/lora/test_olmoe_tp.py
index bbc25cb6b8..c9a4fd6df5 100644
--- a/tests/lora/test_olmoe_tp.py
+++ b/tests/lora/test_olmoe_tp.py
@@ -139,6 +139,7 @@ def test_olmoe_lora_mixed(olmoe_lora_files):
         max_loras=4,
         enforce_eager=True,
         trust_remote_code=True,
+        hf_overrides={"head_dtype": "float32"},
         enable_chunked_prefill=True,
     )