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

Fix failing test

Browse Source
This commit is contained in:
NanoCode012
2023-06-09 00:35:03 +09:00
parent babf0fdb71
commit bfd27ba55e
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/validation.py
View File

@@ -54,7 +54,7 @@ def validate_config(cfg):
"Require cfg.hf_use_auth_token to be True for push_dataset_to_hub" "Require cfg.hf_use_auth_token to be True for push_dataset_to_hub"
) )
if "falcon" in cfg.base_model.lower() and cfg.fsdp: if (cfg.base_model and "falcon" in cfg.base_model.lower()) and cfg.fsdp:
raise ValueError("FSDP is not supported for falcon models") raise ValueError("FSDP is not supported for falcon models")
# TODO # TODO