Spaces:
Running
Running
repetition penalty?
Browse files
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)
|