guoday commited on
Commit
6f2bc47
1 Parent(s): 25b8282

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -64,11 +64,11 @@ def generate(
64
  {"input_ids": input_ids},
65
  streamer=streamer,
66
  max_new_tokens=max_new_tokens,
67
- do_sample=True,
68
  top_p=top_p,
69
  top_k=top_k,
70
  num_beams=1,
71
- temperature=temperature,
72
  repetition_penalty=repetition_penalty,
73
  eos_token_id=32021
74
  )
@@ -92,13 +92,13 @@ chat_interface = gr.ChatInterface(
92
  step=1,
93
  value=DEFAULT_MAX_NEW_TOKENS,
94
  ),
95
- gr.Slider(
96
- label="Temperature",
97
- minimum=0,
98
- maximum=4.0,
99
- step=0.1,
100
- value=0,
101
- ),
102
  gr.Slider(
103
  label="Top-p (nucleus sampling)",
104
  minimum=0.05,
 
64
  {"input_ids": input_ids},
65
  streamer=streamer,
66
  max_new_tokens=max_new_tokens,
67
+ do_sample=False,
68
  top_p=top_p,
69
  top_k=top_k,
70
  num_beams=1,
71
+ # temperature=temperature,
72
  repetition_penalty=repetition_penalty,
73
  eos_token_id=32021
74
  )
 
92
  step=1,
93
  value=DEFAULT_MAX_NEW_TOKENS,
94
  ),
95
+ # gr.Slider(
96
+ # label="Temperature",
97
+ # minimum=0,
98
+ # maximum=4.0,
99
+ # step=0.1,
100
+ # value=0,
101
+ # ),
102
  gr.Slider(
103
  label="Top-p (nucleus sampling)",
104
  minimum=0.05,