Files
OpenLLM/src/openllm/utils/dummy_pt_objects.py
2023-05-28 06:36:06 -07:00

46 lines
992 B
Python

from __future__ import annotations
import typing as t
from ..utils import DummyMetaclass, require_backends
class FlanT5(metaclass=DummyMetaclass):
_backends = ["torch"]
def __init__(self, *args: t.Any, **attrs: t.Any):
require_backends(self, ["torch"])
class DollyV2(metaclass=DummyMetaclass):
_backends = ["torch"]
def __init__(self, *args: t.Any, **attrs: t.Any):
require_backends(self, ["torch"])
class StarCoder(metaclass=DummyMetaclass):
_backends = ["torch"]
def __init__(self, *args: t.Any, **attrs: t.Any):
require_backends(self, ["torch"])
class StableLM(metaclass=DummyMetaclass):
_backends = ["torch"]
def __init__(self, *args: t.Any, **attrs: t.Any):
require_backends(self, ["torch"])
class AutoLLM(metaclass=DummyMetaclass):
_backends = ["torch"]
def __init__(self, *args: t.Any, **attrs: t.Any):
require_backends(self, ["torch"])
MODEL_MAPPING_NAMES = None
MODEL_MAPPING = None