giulio98 commited on
Commit
1e412f5
·
verified ·
1 Parent(s): 163c806

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -577,11 +577,14 @@ def chat_response_stream(message: str, history: list, state: dict, compression_d
577
  streamer=streamer,
578
  use_cache=True,
579
  max_new_tokens=1024,
580
- num_beams=1,
581
- do_sample=False,
 
 
582
  temperature=1.0,
583
- top_p=1.0,
584
- top_k=None,
 
585
  )
586
  t = Thread(target=model.generate, kwargs=generate_kwargs)
587
  t.start()
 
577
  streamer=streamer,
578
  use_cache=True,
579
  max_new_tokens=1024,
580
+ # num_beams=1,
581
+ # do_sample=False,
582
+ # top_p=1.0,
583
+ # top_k=None,
584
  temperature=1.0,
585
+ top_k=64,
586
+ top_p=0.95,
587
+ min_p=0.0
588
  )
589
  t = Thread(target=model.generate, kwargs=generate_kwargs)
590
  t.start()