dvilasuero HF staff commited on
Commit
f16e1d1
·
verified ·
1 Parent(s): f1ce634

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=2, progress=progress)
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