fix

Author: wangzaijun

lightllm/server/core/objs/req.py

@@ -234,15 +234,13 @@ def get_prompt_ids_numpy(self):
         return self.shm_prompt_ids.arr[: self.input_len]
 
     def to_router_rpc_obj(self):
-        if hasattr(self, "multimodal_params"):
-            return (
-                self.request_id,
-                self.index_in_shm_mem,
-                self.multimodal_params,
-                self.sample_params.suggested_dp_index,
-            )
-        else:
-            return (self.request_id, self.index_in_shm_mem, None, self.sample_params.suggested_dp_index)
+        assert hasattr(self, "multimodal_params")
+        return (
+            self.request_id,
+            self.index_in_shm_mem,
+            self.multimodal_params,
+            self.sample_params.suggested_dp_index,
+        )
 
     def can_release(self):
         # 只有管理节点有一个引用

lightllm/server/router/model_infer/infer_batch.py

@@ -320,7 +320,7 @@ def __init__(
         req_id: int,
         req_idx: int,
         shm_index: int,
-        multimodal_params=None,
+        multimodal_params: MultimodalParams,
         vocab_size: int = -1,
         init_prefix_cache: bool = True,
     ):