Update src/distilabel_dataset_generator/apps/sft.py
Browse files
src/distilabel_dataset_generator/apps/sft.py
CHANGED
@@ -54,7 +54,7 @@ def generate_system_prompt(dataset_description, progress=gr.Progress()):
|
|
54 |
|
55 |
def generate_sample_dataset(system_prompt, progress=gr.Progress()):
|
56 |
progress(0.1, desc="Initializing sample dataset generation")
|
57 |
-
result = generate_dataset(system_prompt, num_turns=1, num_rows=
|
58 |
progress(1.0, desc="Sample dataset generated")
|
59 |
return result
|
60 |
|
|
|
54 |
|
55 |
def generate_sample_dataset(system_prompt, progress=gr.Progress()):
|
56 |
progress(0.1, desc="Initializing sample dataset generation")
|
57 |
+
result = generate_dataset(system_prompt, num_turns=1, num_rows=1, progress=progress)
|
58 |
progress(1.0, desc="Sample dataset generated")
|
59 |
return result
|
60 |
|