Spaces:
Runtime error
Runtime error
use cache
Browse files
app.py
CHANGED
|
@@ -47,6 +47,7 @@ def predict(message, history, system_prompt, temperature, max_tokens):
|
|
| 47 |
top_p=0.95,
|
| 48 |
temperature=0.5,
|
| 49 |
max_new_tokens=DEFAULT_MAX_NEW_TOKENS,
|
|
|
|
| 50 |
)
|
| 51 |
t = Thread(target=model.generate, kwargs=generate_kwargs)
|
| 52 |
t.start()
|
|
|
|
| 47 |
top_p=0.95,
|
| 48 |
temperature=0.5,
|
| 49 |
max_new_tokens=DEFAULT_MAX_NEW_TOKENS,
|
| 50 |
+
use_cache=True,
|
| 51 |
)
|
| 52 |
t = Thread(target=model.generate, kwargs=generate_kwargs)
|
| 53 |
t.start()
|