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