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

hotfix for capabilities loading (#1331)

Browse Source
This commit is contained in:
Wing Lian
2024-02-26 14:24:28 -05:00
committed by GitHub
parent d75653407c
commit 7de912e097
2 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -13,7 +13,6 @@ from axolotl.utils.config.models.input.v0_4_1 import (
AxolotlConfigWCapabilities,
AxolotlInputConfig,
)
from axolotl.utils.config.models.internals import GPUCapabilities
from axolotl.utils.dict import DictDefault
from axolotl.utils.models import load_model_config
@@ -197,7 +196,7 @@ def normalize_cfg_datasets(cfg):
cfg.datasets[idx].conversation = "chatml"
def validate_config(cfg: DictDefault, capabilities: Optional[GPUCapabilities] = None):
def validate_config(cfg: DictDefault, capabilities: Optional[dict] = None):
if capabilities:
return DictDefault(
dict(AxolotlConfigWCapabilities(**cfg.to_dict(), capabilities=capabilities))