__init__.py 628 Bytes
Newer Older
1
2
from . import (
    anthropic_llms,
Baber Abbasi's avatar
Baber Abbasi committed
3
    api_models,
4
5
    dummy,
    gguf,
6
    hf_audiolm,
7
    hf_steered,
8
    hf_vlms,
9
    huggingface,
10
    ibm_watsonx_ai,
11
    mamba_lm,
12
    nemo_lm,
13
14
    neuron_optimum,
    openai_completions,
15
    optimum_ipex,
16
    optimum_lm,
17
    sglang_causallms,
18
    sglang_generate_API,
19
20
    textsynth,
    vllm_causallms,
21
    vllm_vlms,
22
23
24
)


haileyschoelkopf's avatar
haileyschoelkopf committed
25
# TODO: implement __all__
Michael Feil's avatar
Michael Feil committed
26
27
28


try:
29
30
31
    # 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
32

33
    huggingface_hub.constants.HF_HUB_ENABLE_HF_TRANSFER = True
Michael Feil's avatar
Michael Feil committed
34
35
except ImportError:
    pass