Spaces:
Paused
Paused
Robin Genolet
commited on
Commit
·
6d70053
1
Parent(s):
07b6cfc
fix: top_k
Browse files
utils/epfl_meditron_utils.py
CHANGED
@@ -34,7 +34,7 @@ def get_llm_response(model_name_or_path, temperature, do_sample, top_p, top_k, m
|
|
34 |
do_sample=do_sample,
|
35 |
temperature=temperature,
|
36 |
top_p=top_p,
|
37 |
-
top_k=
|
38 |
repetition_penalty=repetition_penalty
|
39 |
)
|
40 |
|
|
|
34 |
do_sample=do_sample,
|
35 |
temperature=temperature,
|
36 |
top_p=top_p,
|
37 |
+
top_k=top_k,
|
38 |
repetition_penalty=repetition_penalty
|
39 |
)
|
40 |
|