[NPU] Initialize Ascend config for diffusion workers (vllm-project#4386)

Signed-off-by: KexiongYu <yukexiong1@huawei.com>

Author: Fishermanykx

vllm_omni/diffusion/worker/diffusion_worker.py

@@ -255,6 +255,7 @@ def init_device(self) -> None:
             "Final IR op priority after setting vLLM-Omni overrides: %s", vllm_config.kernel_config.ir_op_priority
         )
         self.vllm_config = vllm_config
+        current_omni_platform.init_diffusion_worker_vllm_config(vllm_config)
 
         # Initialize distributed environment
         with (

vllm_omni/platforms/interface.py

@@ -136,6 +136,11 @@ def get_diffusion_model_runner_cls(cls) -> str:
         """
         return "vllm_omni.diffusion.worker.diffusion_model_runner.DiffusionModelRunner"
 
+    @classmethod
+    def init_diffusion_worker_vllm_config(cls, vllm_config: Any) -> None:
+        """Initialize platform-specific state for diffusion worker VllmConfig."""
+        return None
+
     @classmethod
     def get_torch_device(cls, local_rank: int | None = None) -> torch.device:
         raise NotImplementedError

vllm_omni/platforms/npu/platform.py

@@ -57,6 +57,12 @@ def get_omni_ar_worker_cls(cls) -> str:
     def get_omni_generation_worker_cls(cls) -> str:
         return "vllm_omni.platforms.npu.worker.npu_generation_worker.NPUGenerationWorker"
 
+    @classmethod
+    def init_diffusion_worker_vllm_config(cls, vllm_config: Any) -> None:
+        from vllm_ascend.ascend_config import init_ascend_config
+
+        init_ascend_config(vllm_config)
+
     @classmethod
     def get_default_stage_config_path(cls) -> str:
         return "vllm_omni/platforms/npu/stage_configs"