llm-compressor/src/llmcompressor/__init__.py at b00ca59c75514dc25491d7c8e9e7e461ae7da67b · vllm-project/llm-compressor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
"""
A library for compressing large language models utilizing the latest techniques and
research in the field for both training aware and post training techniques.

The library is designed to be flexible and easy to use on top of
PyTorch and HuggingFace Transformers, allowing for quick experimentation.
"""

# flake8: noqa

from .core import LLMCompressor
from .entrypoints import oneshot, post_train, train
from .logger import LoggerConfig, configure_logger, logger
from .version import (
    __version__,
    build_type,
    version,
    version_base,
    version_build,
    version_major,
    version_minor,
    version_patch,
)

__all__ = [
    "__version__",
    "version_base",
    "build_type",
    "version",
    "version_major",
    "version_minor",
    "version_patch",
    "version_build",
    "configure_logger",
    "logger",
    "LoggerConfig",
    "LLMCompressor",
    "oneshot",
    "post_train",
    "train",
]