14 lines
358 B
Python
14 lines
358 B
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
|
|
class OmniWorkerMixin:
|
|
"""Mixin to ensure Omni plugins are loaded in worker processes."""
|
|
|
|
def __init__(self, *args: Any, **kwargs: Any) -> None:
|
|
super().__init__(*args, **kwargs)
|
|
from vllm_omni.plugins import load_omni_general_plugins
|
|
|
|
load_omni_general_plugins()
|