mirror of
https://github.com/bentoml/OpenLLM.git
synced 2026-01-28 09:21:59 -05:00
11 lines
447 B
Python
11 lines
447 B
Python
from __future__ import annotations
|
|
import typing as t
|
|
from collections import OrderedDict
|
|
from .configuration_auto import CONFIG_MAPPING_NAMES
|
|
from .factory import BaseAutoLLMClass, _LazyAutoMapping
|
|
|
|
MODEL_TF_MAPPING_NAMES = OrderedDict([("flan_t5", "TFFlanT5"), ("opt", "TFOPT")])
|
|
MODEL_TF_MAPPING = _LazyAutoMapping(CONFIG_MAPPING_NAMES, MODEL_TF_MAPPING_NAMES)
|
|
class AutoTFLLM(BaseAutoLLMClass):
|
|
_model_mapping: t.ClassVar = MODEL_TF_MAPPING
|