yonikremer commited on
Commit
49ad8d3
1 Parent(s): 7751341

parameter changed - top k = 1

Browse files
Files changed (1) hide show
  1. hanlde_form_submit.py +1 -2
hanlde_form_submit.py CHANGED
@@ -41,8 +41,7 @@ def create_pipeline(model_name: str, group_size: int) -> GroupedSamplingPipeLine
41
  model_name=model_name,
42
  group_size=group_size,
43
  end_of_sentence_stop=False,
44
- temp=0.5,
45
- top_p=0.6,
46
  )
47
  pipeline_end_time = time()
48
  pipeline_time = pipeline_end_time - pipeline_start_time
 
41
  model_name=model_name,
42
  group_size=group_size,
43
  end_of_sentence_stop=False,
44
+ top_k=1,
 
45
  )
46
  pipeline_end_time = time()
47
  pipeline_time = pipeline_end_time - pipeline_start_time