aryo100 commited on
Commit
18acf6d
·
1 Parent(s): bfee672

update app

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -60,10 +60,12 @@ def generate_stream(prompt, max_new_tokens=128):
60
  # if text.strip():
61
  # yield text
62
 
63
- streamer = TextIteratorStreamer(tokenizer, skip_special_tokens=True)
64
  generation_kwargs = dict(
65
  **inputs,
66
  max_new_tokens=max_new_tokens,
 
 
67
  temperature=0.7,
68
  streamer=streamer
69
  )
 
60
  # if text.strip():
61
  # yield text
62
 
63
+ streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
64
  generation_kwargs = dict(
65
  **inputs,
66
  max_new_tokens=max_new_tokens,
67
+ eos_token_id=tokenizer.eos_token_id,
68
+ do_sample=True,
69
  temperature=0.7,
70
  streamer=streamer
71
  )