[Test][ROCm] Stabilize the mixed OLMoE LoRA test (#58945)
Signed-off-by: Andreas Karatzas <andreas.karatzas@protonmail.com>
Andreas Karatzas committed
Sep 27, 2026 at 22:47 UTC
77fbd9e225a740b54f3281d3bbb1dd54b03d99e5
1 file changed
+1
tests/lora/test_olmoe_tp.py
+1
@@ -139,6 +139,7 @@ def test_olmoe_lora_mixed(olmoe_lora_files):
139
max_loras=4,
140
enforce_eager=True,
141
trust_remote_code=True,
142
+ hf_overrides={"head_dtype": "float32"},
143
enable_chunked_prefill=True,
144
)
145