Spaces:
Runtime error
Runtime error
Increase batch size
Browse files
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,
|