mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-28 03:43:01 +00:00
91fa553775
* [feature] qlora support * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * qlora follow commit * migrate qutization folder to colossalai/ * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * minor fixes --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
8 lines
144 B
Python
8 lines
144 B
Python
from .bnb import quantize_model
|
|
from .bnb_config import BnbQuantizationConfig
|
|
|
|
__all__ = [
|
|
"BnbQuantizationConfig",
|
|
"quantize_model",
|
|
]
|