diff --git a/test/registered/8-gpu-models/test_mimo_models.py b/test/registered/8-gpu-models/test_mimo_models.py index e2651ad30..6d0cc0899 100644 --- a/test/registered/8-gpu-models/test_mimo_models.py +++ b/test/registered/8-gpu-models/test_mimo_models.py @@ -45,7 +45,7 @@ class TestMiMoV2Flash(GSM8KMixin, SpecDecodingMixin, DefaultServerBase): ] bs_1_speed_thres = 170 - accept_length_thres = 3.2 + num_accepted_drafts_thres = 3.2 if __name__ == "__main__":