Spaces:
Running
Running
Update rag.py
Browse files
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)
|