fix bug for waiting queue.

hiworldwzj · web-flow · commit aa0aac882980 · 2025-01-16T12:40:42.000+08:00
diff --git a/lightllm/server/router/req_queue/__init__.py b/lightllm/server/router/req_queue/__init__.py
@@ -21,8 +21,9 @@ def build_req_queue(args, router, dp_size: int):
         queue_class = ContinuesBatchQueue
     if args.first_token_constraint_mode:
         queue_class = ContinuesBatchQueue
-    queue_class = ContinuesBatchQueue
-
+    if queue_class is None:
+        queue_class = ContinuesBatchQueue
+    
     if dp_size == 1:
         return queue_class(args, router, 0, dp_size)
     else:
