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

Add cfg.lora_target_linear

Browse Source
This commit is contained in:
NanoCode012
2023-05-26 14:32:30 +09:00
parent bbfc333a01
commit 919623793a
2 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -364,14 +364,18 @@ def load_lora(model, cfg):
PeftModel,
)
bits = None
if cfg.load_in_4bit:
bits = 4
elif cfg.load_in_8bit:
bits = 8
linear_names = find_all_linear_names(bits, model)
logging.info(f"found linear modules: {repr(linear_names)}")
lora_target_modules = list(set(list(cfg.lora_target_modules) + linear_names))
lora_target_modules = list(cfg.lora_target_modules)
if cfg.lora_target_linear:
bits = None
if cfg.load_in_4bit:
bits = 4
elif cfg.load_in_8bit:
bits = 8
linear_names = find_all_linear_names(bits, model)
logging.info(f"found linear modules: {repr(linear_names)}")
lora_target_modules = list(set(lora_target_modules + linear_names))
lora_config = LoraConfig(
r=cfg.lora_r,