16 lines
435 B
Python
16 lines
435 B
Python
# 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",
|
|
]
|