cszhzleo commited on
Commit
3854982
1 Parent(s): 0d0b874

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -44,6 +44,7 @@ parameters = {
44
  "max_new_tokens": 768,
45
  "repetition_penalty": 1.2,
46
  "return_full_text": False,
 
47
  }
48
 
49
 
 
44
  "max_new_tokens": 768,
45
  "repetition_penalty": 1.2,
46
  "return_full_text": False,
47
+ "stop": ["</s>"],
48
  }
49
 
50