dvilasuero HF staff commited on
Commit
0b6ba72
1 Parent(s): 3e29cb8

More fixes

Browse files
src/distilabel_dataset_generator/pipelines/sft.py CHANGED
@@ -222,8 +222,8 @@ def get_pipeline(num_turns, num_rows, system_prompt):
222
  "temperature": 0.8,
223
  "max_new_tokens": 1024
224
  },
225
- output_mappings={"generation": "completion"}
226
- )
227
  )
228
 
229
  keep_columns = KeepColumns(
 
222
  "temperature": 0.8,
223
  "max_new_tokens": 1024
224
  },
225
+ ),
226
+ output_mappings={"generation": "completion"}
227
  )
228
 
229
  keep_columns = KeepColumns(