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

2024-04-08 21:37:54 -04:00
parent 2fa65b9599
commit ff01c45127
1 changed files with 1 additions and 0 deletions
--- 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