optimum/amd/__init__.py

# Copyright 2023 The HuggingFace Team. All rights reserved.
# Licensed under the MIT License.

from typing import TYPE_CHECKING

from transformers.utils import _LazyModule


_import_structure = {
    "brevitas.configuration": [
        "BrevitasQuantizationConfig",
    ],
    "brevitas.quantizer": [
        "BrevitasQuantizer",
    ],
}

if TYPE_CHECKING:
    from .brevitas.configuration import BrevitasQuantizationConfig
    from .brevitas.quantizer import BrevitasQuantizer
else:
    import sys

    sys.modules[__name__] = _LazyModule(
        __name__,
        globals()["__file__"],
        _import_structure,
        module_spec=__spec__,
    )