yonikremer commited on
Commit
98c1d3b
1 Parent(s): d00144e

changed top-k to 50

Browse files
Files changed (1) hide show
  1. hanlde_form_submit.py +1 -1
hanlde_form_submit.py CHANGED
@@ -38,7 +38,7 @@ def create_pipeline(model_name: str, group_size: int) -> GroupedSamplingPipeLine
38
  model_name=model_name,
39
  group_size=group_size,
40
  end_of_sentence_stop=False,
41
- top_k=1,
42
  )
43
  pipeline_end_time = time()
44
  pipeline_time = pipeline_end_time - pipeline_start_time
 
38
  model_name=model_name,
39
  group_size=group_size,
40
  end_of_sentence_stop=False,
41
+ top_k=50,
42
  )
43
  pipeline_end_time = time()
44
  pipeline_time = pipeline_end_time - pipeline_start_time