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