Commit
·
0c58a58
1
Parent(s):
55add14
fix: add token to run pipeline
Browse files
src/distilabel_dataset_generator/sft.py
CHANGED
|
@@ -181,7 +181,8 @@ def _generate_dataset(
|
|
| 181 |
gr.Info("Started pipeline execution.")
|
| 182 |
result_queue = multiprocessing.Queue()
|
| 183 |
p = multiprocessing.Process(
|
| 184 |
-
target=_run_pipeline,
|
|
|
|
| 185 |
)
|
| 186 |
p.start()
|
| 187 |
p.join()
|
|
|
|
| 181 |
gr.Info("Started pipeline execution.")
|
| 182 |
result_queue = multiprocessing.Queue()
|
| 183 |
p = multiprocessing.Process(
|
| 184 |
+
target=_run_pipeline,
|
| 185 |
+
args=(result_queue, _num_turns, _num_rows, _system_prompt, _token.token),
|
| 186 |
)
|
| 187 |
p.start()
|
| 188 |
p.join()
|