__init__.py 508 Bytes
Newer Older
1
2
3
4
from . import (
    anthropic_llms,
    dummy,
    gguf,
haileyschoelkopf's avatar
haileyschoelkopf committed
5
    hf_vlms,
6
    huggingface,
Ashvin Nihalani's avatar
Ashvin Nihalani committed
7
    llava,
8
    mamba_lm,
9
    nemo_lm,
10
    neuralmagic,
11
12
13
14
15
16
17
18
    neuron_optimum,
    openai_completions,
    optimum_lm,
    textsynth,
    vllm_causallms,
)


haileyschoelkopf's avatar
haileyschoelkopf committed
19
# TODO: implement __all__
Michael Feil's avatar
Michael Feil committed
20
21
22


try:
23
24
25
    # enable hf hub transfer if available
    import hf_transfer  # type: ignore # noqa
    import huggingface_hub.constants  # type: ignore
Michael Feil's avatar
Michael Feil committed
26

27
    huggingface_hub.constants.HF_HUB_ENABLE_HF_TRANSFER = True
Michael Feil's avatar
Michael Feil committed
28
29
except ImportError:
    pass