dvilasuero HF staff commited on
Commit
f16cf7d
1 Parent(s): aefe980

input mappings again

Browse files
src/distilabel_dataset_generator/pipelines/sft.py CHANGED
@@ -224,7 +224,7 @@ def get_pipeline(num_turns, num_rows, system_prompt):
224
  },
225
  ),
226
  output_mappings={"generation": "completion"},
227
- input_mappings={"prompt": "instruction"}
228
  )
229
 
230
  keep_columns = KeepColumns(
 
224
  },
225
  ),
226
  output_mappings={"generation": "completion"},
227
+ input_mappings={"instruction": "prompt"}
228
  )
229
 
230
  keep_columns = KeepColumns(