JoPmt commited on
Commit
0d161e8
·
1 Parent(s): 22e431a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def predict(model: str,inputs: str,typical_p: float,top_p: float,temperature: fl
39
  partial_words = ""
40
 
41
  if model in ("bigcode/starcoder", "bigcode/starcoder"):
42
- iterator = client.generate_stream(total_inputs,typical_p=typical_p,truncate=1024,watermark=False,max_new_tokens=1024,)
43
 
44
  for i, response in enumerate(iterator):
45
  if response.token.special:
 
39
  partial_words = ""
40
 
41
  if model in ("bigcode/starcoder", "bigcode/starcoder"):
42
+ iterator = client.generate_stream(total_inputs,typical_p=typical_p,truncate=300,watermark=False,max_new_tokens=299,)
43
 
44
  for i, response in enumerate(iterator):
45
  if response.token.special: