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 |
-
|
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(
|