[minor, fix] Allow passing class interface as custom pipeline argument (#2973)

Signed-off-by: knlnguyen1802 <knlnguyen1802@gmail.com>
Signed-off-by: Samit <285365963@qq.com>
Co-authored-by: Samit <285365963@qq.com>

vllm_omni/diffusion/worker/diffusion_model_runner.py

@@ -102,7 +102,7 @@ class DiffusionModelRunner(OmniConnectorModelRunnerMixin):
         self,
         memory_pool_context_fn: callable | None = None,
         load_format: str | None = None,
-        custom_pipeline_name: str | None = None,
+        custom_pipeline_name: str | type | None = None,
     ) -> None:
         """
         Load the diffusion model, apply compilation and offloading.