fb5ec0dc9e
Signed-off-by: Woosuk Kwon <woosuk@inferact.ai> Co-authored-by: Bugen Zhao <i@bugenzhao.com> Co-authored-by: Giancarlo Delfin <32987265+TheEpicDolphin@users.noreply.github.com> Co-authored-by: Isotr0py <Isotr0py@outlook.com> Co-authored-by: Isotr0py <mozf@inferact.ai> Co-authored-by: Jee Jee Li <jeejeelee@inferact.ai> Co-authored-by: Roger Wang <hey@rogerw.io> Co-authored-by: Yifan Qiao <yifanqiao@inferact.ai> Co-authored-by: Claude Fable 5 <noreply@anthropic.com> Co-authored-by: OpenAI Codex <codex@openai.com>
29 lines
695 B
Python
29 lines
695 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from .nvidia.model import (
|
|
InklingForCausalLM,
|
|
InklingForConditionalGeneration,
|
|
)
|
|
from .nvidia.mtp import InklingMTP
|
|
|
|
__all__ = [
|
|
"InklingForConditionalGeneration",
|
|
"InklingForCausalLM",
|
|
"InklingMTP",
|
|
]
|
|
|
|
|
|
def __getattr__(name: str):
|
|
if name == "InklingMTP":
|
|
from .nvidia import mtp
|
|
|
|
return mtp.InklingMTP
|
|
if name in __all__:
|
|
from .nvidia import model
|
|
|
|
return getattr(model, name)
|
|
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|