From ff01c45127aee0eb607afe20ea5cde78f67e0791 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 8 Apr 2024 21:37:54 -0400 Subject: [PATCH] add field to sft dataset pydantic for completion support (#1497) --- src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py index 6b41db6b05..20887dccf8 100644 --- a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py +++ b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py @@ -98,6 +98,7 @@ class SFTDataset(BaseModel): ds_type: Optional[str] = None train_on_split: Optional[str] = None + field: Optional[str] = None field_human: Optional[str] = None field_model: Optional[str] = None