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

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -52,7 +52,7 @@ class RAG:
52
  {"role": "user", "content": query}
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)
 
52
  {"role": "user", "content": query}
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)