__init__.py 441 Bytes
Newer Older
1
from . import huggingface
2
from . import openai_completions
Jonathan Tow's avatar
Jonathan Tow committed
3
from . import textsynth
Leo Gao's avatar
Leo Gao committed
4
from . import dummy
5
from . import anthropic_llms
6
from . import gguf
Hailey Schoelkopf's avatar
Hailey Schoelkopf committed
7
from . import vllm_causallms
8
from . import mamba_lm
9
from . import optimum_lm
10
from . import neuron_optimum
haileyschoelkopf's avatar
haileyschoelkopf committed
11
# TODO: implement __all__
Michael Feil's avatar
Michael Feil committed
12
13
14
15
16
17
18
19
20
21
22


import os

try:
    # enabling faster model download
    import hf_transfer

    os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
except ImportError:
    pass