Robin Genolet commited on
Commit
6d70053
·
1 Parent(s): 07b6cfc

fix: top_k

Browse files
Files changed (1) hide show
  1. utils/epfl_meditron_utils.py +1 -1
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=top_p,
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