diff --git a/tests/model/test_qwen3_5.py b/tests/model/test_qwen3_5.py index 8af0cfde0..df695af1a 100644 --- a/tests/model/test_qwen3_5.py +++ b/tests/model/test_qwen3_5.py @@ -300,7 +300,7 @@ def test_qwen3_5_vl_run_mtp(self, device, sp_size, tol): atol=tol, rtol=tol ), - f"Expected text loss around {key}, but got {loss.item()}" + f"Expected {key} loss around {loss_reference[key]}, but got {loss.item()}" )