Simplify tests & Fix trtllm custom allreduce registration (#4252)

2025-03-10 01:24:22 -07:00
parent 007f8b3dc2
commit aa957102a9
13 changed files with 30 additions and 211 deletions
--- a/test/srt/test_gptqmodel_dynamic.py
+++ b/test/srt/test_gptqmodel_dynamic.py
@@ -129,6 +129,7 @@ class TestGPTQModelDynamic(unittest.TestCase):
                "text": "The capital of France is",
                "sampling_params": {
                    "max_new_tokens": max_new_tokens,
+                    "temperature": 0.001,
                },
            },
        )