maknee commited on
Commit
f62f025
·
1 Parent(s): 5a903b2

Increase batch size

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,7 +33,8 @@ async def chat(history, limit: int = 1024, temp: float = 0.8, top_k: int = 40, t
33
  temperature = float(temp),
34
  top_k = int(top_k),
35
  top_p = float(top_p),
36
- repetition_penalty = float(repeat_penalty)
 
37
  )
38
  ai = Kani(
39
  engine,
 
33
  temperature = float(temp),
34
  top_k = int(top_k),
35
  top_p = float(top_p),
36
+ repetition_penalty = float(repeat_penalty),
37
+ batch_size = 512,
38
  )
39
  ai = Kani(
40
  engine,