vllm-omni/vllm_omni/diffusion/worker/__init__.py

# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
"""Worker classes for diffusion models."""

from vllm_omni.diffusion.worker.gpu_diffusion_model_runner import GPUDiffusionModelRunner
from vllm_omni.diffusion.worker.gpu_diffusion_worker import (
    GPUDiffusionWorker,
    WorkerProc,
)

__all__ = [
    "GPUDiffusionModelRunner",
    "GPUDiffusionWorker",
    "WorkerProc",
]