tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix auto linear modules for lora w/o any set already

Browse Source
This commit is contained in:
Wing Lian
2023-05-27 08:49:43 -04:00
parent 4c500f5830
commit 4c906339f7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/models.py
View File

@@ -364,7 +364,7 @@ def load_lora(model, cfg):
PeftModel, PeftModel,
) )
lora_target_modules = list(cfg.lora_target_modules) lora_target_modules = list(cfg.lora_target_modules or [])
if cfg.lora_target_linear: if cfg.lora_target_linear:
bits = None bits = None