diff --git a/vllm_omni/engine/arg_utils.py b/vllm_omni/engine/arg_utils.py index f2adca975..e7ef9f603 100644 --- a/vllm_omni/engine/arg_utils.py +++ b/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,