HFHAB commited on
Commit
727f564
·
verified ·
1 Parent(s): 18a95a6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -15,7 +15,7 @@ class Item(BaseModel):
15
  temperature: float = 0.3
16
  max_new_tokens: int = 3000
17
  top_p: float = 0.15
18
- repetition_penalty: float = 1.5
19
 
20
  def format_prompt(message, history):
21
  prompt = "<s>"
 
15
  temperature: float = 0.3
16
  max_new_tokens: int = 3000
17
  top_p: float = 0.15
18
+ repetition_penalty: float = 1.0
19
 
20
  def format_prompt(message, history):
21
  prompt = "<s>"