davidberenstein1957 HF staff commited on
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, args=(result_queue, _num_turns, _num_rows, _system_prompt)
 
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()