Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -12,3 +12,4 @@ tmp_autoround/
ut_log_dir/
CLAUDE.local.md
docs/plan/
.worktrees/
28 changes: 28 additions & 0 deletions auto_round/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,15 @@

# support for old api
from auto_round.autoround import AutoRoundLLM, AutoRoundMLLM, AutoRoundAdam, AutoRoundDiffusion
from auto_round.algorithms.quantization.rtn.config import OptimizedRTNConfig, RTNConfig
from auto_round.algorithms.quantization.sign_round.config import (
AdamRoundConfig,
SignRoundConfig,
SignRoundV2Config,
)
from auto_round.algorithms.transforms.awq.config import AWQConfig
from auto_round.algorithms.transforms.quarot.config import RotationConfig
from auto_round.algorithms.transforms.spinquant.preprocessor import SpinQuantConfig
from auto_round.schemes import QuantizationScheme
from auto_round.auto_scheme import AutoScheme
from auto_round.utils import LazyImport
Expand All @@ -23,3 +32,22 @@
monkey_patch()

from .version import __version__

__all__ = [
"__version__",
"AutoRound",
"AutoRoundLLM",
"AutoRoundMLLM",
"AutoRoundAdam",
"AutoRoundDiffusion",
"AutoScheme",
"QuantizationScheme",
"RTNConfig",
"OptimizedRTNConfig",
"SignRoundConfig",
"AdamRoundConfig",
"SignRoundV2Config",
"AWQConfig",
"RotationConfig",
"SpinQuantConfig",
]
Loading
Loading