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

fix(log): improve warning to clarify that lora_modules_to_save expect a list (#1197)

Browse Source
This commit is contained in:
NanoCode012
2024-01-25 10:08:34 +09:00
committed by GitHub
parent 1427d5b502
commit 08719b9609
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -67,7 +67,7 @@ def check_model_config(cfg: DictDefault, model_config: Union[AutoConfig, DictDef
): ):
lora_modules_to_save = ", ".join(map(lambda x: f"`{x}`", lora_modules_to_save)) lora_modules_to_save = ", ".join(map(lambda x: f"`{x}`", lora_modules_to_save))
raise ValueError( raise ValueError(
f"`lora_modules_to_save` not properly set when adding new tokens. Please include {lora_modules_to_save} in `lora_modules_to_save`." f"`lora_modules_to_save` not properly set when adding new tokens. Please include [{lora_modules_to_save}] in `lora_modules_to_save`."
) )
@@ -182,7 +182,7 @@ def load_tokenizer(cfg):
[f"`{x}`" for x in lora_modules_to_save] [f"`{x}`" for x in lora_modules_to_save]
) )
raise ValueError( raise ValueError(
f"Please set lora_modules_to_save to {lora_modules_to_save} when using an adapter and changing the special tokens." f"Please set lora_modules_to_save to [{lora_modules_to_save}] when using an adapter and changing the special tokens."
) )
tokenizer.add_special_tokens( tokenizer.add_special_tokens(