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


Rayyyyy's avatar
Rayyyyy committed
17
# TODO: implement __all__
Rayyyyy's avatar
Rayyyyy committed
18

Rayyyyy's avatar
Rayyyyy committed
19
20
21
22
23
24
25
26
27

try:
    # enable hf hub transfer if available
    import hf_transfer  # type: ignore # noqa
    import huggingface_hub.constants  # type: ignore

    huggingface_hub.constants.HF_HUB_ENABLE_HF_TRANSFER = True
except ImportError:
    pass