[Test][ROCm] Stabilize the mixed OLMoE LoRA test (#58945)

Signed-off-by: Andreas Karatzas <andreas.karatzas@protonmail.com>
2026-09-28 11:47:41 +08:00
parent 5eaa50f151
commit 77fbd9e225
1 changed files with 1 additions and 0 deletions
@@ -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,
    )