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