Update __init__.py

This commit is contained in:
Kohaku-Blueleaf 2025-07-22 16:33:45 +08:00
parent 545cb2187c
commit bf053a9c55

View File

@ -19,11 +19,13 @@ adapter_maps: dict[str, type[WeightAdapterBase]] = {
"LoRA": LoRAAdapter,
"LoHa": LoHaAdapter,
"LoKr": LoKrAdapter,
"GLoRA": GLoRAAdapter,
"OFT": OFTAdapter,
"BOFT": BOFTAdapter,
## We disable not implemented algo for now
# "GLoRA": GLoRAAdapter,
# "OFT": OFTAdapter,
# "BOFT": BOFTAdapter,
}
__all__ = [
"WeightAdapterBase",
"WeightAdapterTrainBase",