| # 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__"] | |