michailroussos commited on
Commit
80bc875
·
1 Parent(s): db497f0
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,13 +36,13 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
36
 
37
  # Use a TextStreamer for real-time decoding
38
  streamer = TextStreamer(tokenizer, skip_prompt=True)
39
- model.generate(
40
  input_ids=inputs,
41
- streamer=streamer,
42
  max_new_tokens=max_tokens,
43
  use_cache=True,
44
  temperature=temperature,
45
  top_p=top_p,
 
46
  )
47
 
48
 
 
36
 
37
  # Use a TextStreamer for real-time decoding
38
  streamer = TextStreamer(tokenizer, skip_prompt=True)
39
+ _ = model.generate(
40
  input_ids=inputs,
 
41
  max_new_tokens=max_tokens,
42
  use_cache=True,
43
  temperature=temperature,
44
  top_p=top_p,
45
+ streamer=streamer,
46
  )
47
 
48