merge main

Signed-off-by: Taichang Zhou <tzhouam@connect.ust.hk>

vllm_omni/engine/arg_utils.py

@@ -193,6 +193,7 @@ class OmniEngineArgs(EngineArgs):
             async_chunk=self.async_chunk,
             model_stage=self.model_stage,
             model_arch=self.model_arch,
+            worker_type=self.worker_type,
             engine_output_type=self.engine_output_type,
             hf_config_name=self.hf_config_name,
             custom_process_next_stage_input_func=self.custom_process_next_stage_input_func,
@@ -350,6 +351,7 @@ class AsyncOmniEngineArgs(AsyncEngineArgs):
             async_chunk=self.async_chunk,
             model_stage=self.model_stage,
             model_arch=self.model_arch,
+            worker_type=self.worker_type,
             engine_output_type=self.engine_output_type,
             hf_config_name=self.hf_config_name,
             custom_process_next_stage_input_func=self.custom_process_next_stage_input_func,