pvduy commited on
Commit
5251b14
·
1 Parent(s): 5081c38
Files changed (1) hide show
  1. app.py +1 -0
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()