Spaces:

argilla
/

synthetic-data-generator

Running

sdiazlor HF staff commited on 23 days ago

Commit

a3f4be7

•

1 Parent(s): 46f00bc

fix: typo

Files changed (1) hide show

src/distilabel_dataset_generator/pipelines/textcat.py CHANGED Viewed

@@ -155,7 +155,6 @@ with Pipeline(name="textcat") as pipeline:
             generation_kwargs={{
                 "temperature": 0.8,
                 "max_new_tokens": 2048,
-                "do_sample": True,
             }},
         ),
         n={num_labels},

             generation_kwargs={{
                 "temperature": 0.8,
                 "max_new_tokens": 2048,
             }},
         ),
         n={num_labels},