add field to sft dataset pydantic for completion support (#1497)

axolotl-ai-cloud · Apr 9, 2024 · ff01c45 · ff01c45
1 parent 2fa65b9
commit ff01c45
Showing 1 changed file with 1 addition and 0 deletions.
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
@@ -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