Skip to content

text

Text-generation-specific model constants, enums, and descriptors.

_TEXT_BACKEND_REGISTRY module-attribute 

_TEXT_BACKEND_REGISTRY = EnumRegistry(
    (value) for item in TEXT_BACKENDS
)

KNOWN_TEXT_BACKENDS module-attribute 

KNOWN_TEXT_BACKENDS = mutable_values()

TEXT_BACKENDS

Bases: StrEnum

An enum of all the text backends.

Source code in src/horde_model_reference/model_consts/text.py 
10
11
12
13
14
class TEXT_BACKENDS(StrEnum):
    """An enum of all the text backends."""

    aphrodite = auto()
    koboldcpp = auto()

aphrodite class-attribute instance-attribute 

aphrodite = auto()

koboldcpp class-attribute instance-attribute 

koboldcpp = auto()

register_text_backend 

register_text_backend(backend: str) -> None

Register a new text backend.

Parameters:

  • backend (str) –

    The text backend to register.

Source code in src/horde_model_reference/model_consts/text.py 
21
22
23
24
25
26
27
28
def register_text_backend(backend: str) -> None:
    """Register a new text backend.

    Args:
        backend: The text backend to register.

    """
    _TEXT_BACKEND_REGISTRY.register(backend)

is_known_text_backend 

is_known_text_backend(backend: str) -> bool

Check if a text backend is known.

Parameters:

  • backend (str) –

    The text backend to check.

Returns:

  • bool –

    True if the text backend is known, False otherwise.

Source code in src/horde_model_reference/model_consts/text.py 
31
32
33
34
35
36
37
38
39
40
41
def is_known_text_backend(backend: str) -> bool:
    """Check if a text backend is known.

    Args:
        backend: The text backend to check.

    Returns:
        True if the text backend is known, False otherwise.

    """
    return _TEXT_BACKEND_REGISTRY.is_known(backend)