Spaces:

argilla
/

synthetic-data-generator

Running

dvilasuero HF staff commited on Sep 11

Commit

cb13241

•

1 Parent(s): f16cf7d

Update src/distilabel_dataset_generator/pipelines/sft.py

Files changed (1) hide show

src/distilabel_dataset_generator/pipelines/sft.py CHANGED Viewed

@@ -223,6 +223,7 @@ def get_pipeline(num_turns, num_rows, system_prompt):
                         "max_new_tokens": 1024
                     },
                 ),
                 output_mappings={"generation": "completion"},
                 input_mappings={"instruction": "prompt"}
             )

                         "max_new_tokens": 1024
                     },
                 ),
+                system_prompt=system_prompt,
                 output_mappings={"generation": "completion"},
                 input_mappings={"instruction": "prompt"}
             )