8 lines
262 B
Python
8 lines
262 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_worker import GPUWorker, WorkerProc
|
|
|
|
__all__ = ["GPUWorker", "WorkerProc"]
|