winglian commited on
Commit
ff01c45
1 Parent(s): 2fa65b9

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

Browse files
src/axolotl/utils/config/models/input/v0_4_1/__init__.py CHANGED
@@ -98,6 +98,7 @@ class SFTDataset(BaseModel):
98
  ds_type: Optional[str] = None
99
  train_on_split: Optional[str] = None
100
 
 
101
  field_human: Optional[str] = None
102
  field_model: Optional[str] = None
103
 
 
98
  ds_type: Optional[str] = None
99
  train_on_split: Optional[str] = None
100
 
101
+ field: Optional[str] = None
102
  field_human: Optional[str] = None
103
  field_model: Optional[str] = None
104