File size: 295 Bytes
6f7a484 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | from __future__ import annotations
from vllm import ModelRegistry
from .constants import ARCHITECTURE
def register() -> None:
ModelRegistry.register_model(
ARCHITECTURE,
"kalm_t5gemma2_vllm_plugin.modeling:T5Gemma2VllmScoreClassification",
)
__all__ = ["register"]
|