erfanzar commited on
Commit
c020e09
1 Parent(s): f4950d1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -31,11 +31,11 @@ LinguaMatic utilizes the llama2 prompting method to generate responses. This met
31
  ```python
32
  def llama_prompt(
33
  message: str,
34
- chat_history,
35
- system: str
36
  ) -> str:
37
  do_strip = False
38
- texts = [f"<s>[INST] <<SYS>>\n{system}\n<</SYS>>\n\n"] if system is not None else "<s>[INST] "
39
  for user_input, response in chat_history:
40
  user_input = user_input.strip() if do_strip else user_input
41
  do_strip = True
 
31
  ```python
32
  def llama_prompt(
33
  message: str,
34
+ chat_history: list = None,
35
+ system: str = None
36
  ) -> str:
37
  do_strip = False
38
+ texts = [f"<s>[INST] <<SYS>>\n{system}\n<</SYS>>\n\n"] if system is not None else ["<s>[INST] "]
39
  for user_input, response in chat_history:
40
  user_input = user_input.strip() if do_strip else user_input
41
  do_strip = True