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

hotfix to exclude_unset from pydantic config when converting back to a dict (#1334)

Browse Source
This commit is contained in:
Wing Lian
2024-02-26 15:06:25 -05:00
committed by GitHub
parent e7eed203d8
commit 269c5436ea
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/axolotl/utils/config/__init__.py
View File

@@ -199,9 +199,15 @@ def normalize_cfg_datasets(cfg):
def validate_config(cfg: DictDefault, capabilities: Optional[dict] = None): def validate_config(cfg: DictDefault, capabilities: Optional[dict] = None):
if capabilities: if capabilities:
return DictDefault( return DictDefault(
dict(AxolotlConfigWCapabilities(**cfg.to_dict(), capabilities=capabilities)) dict(
AxolotlConfigWCapabilities(
**cfg.to_dict(), capabilities=capabilities
).model_dump(exclude_unset=True)
)
) )
return DictDefault(dict(AxolotlInputConfig(**cfg.to_dict()))) return DictDefault(
dict(AxolotlInputConfig(**cfg.to_dict()).model_dump(exclude_unset=True))
)
def legacy_validate_config(cfg): def legacy_validate_config(cfg):