File size: 652 Bytes
49dddf5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | # Copyright 2026 Angshul Majumdar. Licensed under the Apache License, Version 2.0.
"""TensorPEFT: parameter-efficient fine-tuning with tensor gradient-pursuit adapters."""
__version__ = "0.1.0"
from .config import TensorPEFTConfig # noqa: E402
from .model import TensorPEFTModel, get_tensorpeft_model # noqa: E402
def __getattr__(name): # lazy: importing the Trainer pulls in accelerate
if name == "TensorPEFTTrainer":
from .trainer import TensorPEFTTrainer
return TensorPEFTTrainer
raise AttributeError(name)
__all__ = ["TensorPEFTConfig", "TensorPEFTModel", "TensorPEFTTrainer", "get_tensorpeft_model", "__version__"]
|