huedaya commited on
Commit
f441c05
·
verified ·
1 Parent(s): 1930354

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -6
main.py CHANGED
@@ -18,12 +18,7 @@ class Item(BaseModel):
18
  repetition_penalty: float = 1.0
19
 
20
  def format_prompt(message, history):
21
- prompt = "<s>"
22
- for user_prompt, bot_response in history:
23
- prompt += f"[INST] {user_prompt} [/INST]"
24
- prompt += f" {bot_response}</s> "
25
- prompt += f"[INST] {message} [/INST]"
26
- return prompt
27
 
28
  def generate(item: Item):
29
  temperature = float(item.temperature)
 
18
  repetition_penalty: float = 1.0
19
 
20
  def format_prompt(message, history):
21
+ return message
 
 
 
 
 
22
 
23
  def generate(item: Item):
24
  temperature = float(item.temperature)