Spaces:
Running
Running
Update rag.py
Browse files
rag.py
CHANGED
@@ -55,7 +55,7 @@ class RAG:
|
|
55 |
# presence_penalty=model_parameters['repetition_penalty'],
|
56 |
# extra_body=model_parameters,
|
57 |
stream=False,
|
58 |
-
stop=["<|im_end|>", "<|end_header_id|>", "<|eot_id|>", "<|reserved_special_token
|
59 |
)
|
60 |
return(chat_completion.choices[0].message.content)
|
61 |
|
|
|
55 |
# presence_penalty=model_parameters['repetition_penalty'],
|
56 |
# extra_body=model_parameters,
|
57 |
stream=False,
|
58 |
+
stop=["<|im_end|>", "<|end_header_id|>", "<|eot_id|>", "<|reserved_special_token"]
|
59 |
)
|
60 |
return(chat_completion.choices[0].message.content)
|
61 |
|