jgrivolla commited on
Commit
853b97c
·
verified ·
1 Parent(s): b4c71d5

repetition penalty?

Browse files
Files changed (1) hide show
  1. rag.py +1 -0
rag.py CHANGED
@@ -53,6 +53,7 @@ class RAG:
53
  ],
54
  max_tokens=model_parameters['max_new_tokens'], # TODO: map other parameters
55
  frequency_penalty=model_parameters['repetition_penalty'],
 
56
  stream=False
57
  )
58
  print(chat_completion)
 
53
  ],
54
  max_tokens=model_parameters['max_new_tokens'], # TODO: map other parameters
55
  frequency_penalty=model_parameters['repetition_penalty'],
56
+ repetition_penalty=model_parameters['repetition_penalty'],
57
  stream=False
58
  )
59
  print(chat_completion)