dvilasuero HF staff commited on
Commit
bd0983f
1 Parent(s): bddcd7b

fix: add batch size 1 for pipe code

Browse files
src/distilabel_dataset_generator/pipelines/sft.py CHANGED
@@ -175,6 +175,7 @@ with Pipeline(name="sft") as pipeline:
175
  ),
176
  n_turns={num_turns},
177
  num_rows={num_rows},
 
178
  system_prompt=SYSTEM_PROMPT,
179
  output_mappings={input_mappings},
180
  )
 
175
  ),
176
  n_turns={num_turns},
177
  num_rows={num_rows},
178
+ batch_size=1,
179
  system_prompt=SYSTEM_PROMPT,
180
  output_mappings={input_mappings},
181
  )