ankush13r commited on
Commit
3bb55c2
·
verified ·
1 Parent(s): dc8f8ea

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
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