xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 11:17:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add adapter map for algo selection

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2025-07-22 16:29:32 +08:00
parent f5bdfcf0fe
commit 7aa3184e85
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
comfy/weight_adapter/__init__.py
View File

@ -15,9 +15,18 @@ adapters: list[type[WeightAdapterBase]] = [
OFTAdapter,
BOFTAdapter,
]
adapter_maps: dict[str, type[WeightAdapterBase]] = {
"LoRA": LoRAAdapter,
"LoHa": LoHaAdapter,
"LoKr": LoKrAdapter,
"GLoRA": GLoRAAdapter,
"OFT": OFTAdapter,
"BOFT": BOFTAdapter,
}
__all__ = [
"WeightAdapterBase",
"WeightAdapterTrainBase",
"adapters"
"adapters",
"adapter_maps",
] + [a.__name__ for a in adapters]