BoxzDev commited on
Commit
78215e4
·
verified ·
1 Parent(s): bad0ab0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -36,7 +36,7 @@ rejection_responses = [
36
  class Item(BaseModel):
37
  input: str
38
  temperature: float = 0.7
39
- max_new_tokens: int = 1048
40
  top_p: float = 0.9
41
  repetition_penalty: float = 1.1
42
 
 
36
  class Item(BaseModel):
37
  input: str
38
  temperature: float = 0.7
39
+ max_new_tokens: int = 100
40
  top_p: float = 0.9
41
  repetition_penalty: float = 1.1
42