yonikremer
commited on
Commit
•
49ad8d3
1
Parent(s):
7751341
parameter changed - top k = 1
Browse files- 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 |
-
|
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
|