tomasmcm commited on
Commit
ba66f05
·
verified ·
1 Parent(s): cab7e8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -14,7 +14,7 @@ def format_prompt(system, message, history):
14
  return prompt
15
 
16
  def generate(
17
- prompt, history, system_prompt, temperature=0.8, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
18
  ):
19
  temperature = float(temperature)
20
  if temperature < 1e-2:
@@ -50,7 +50,7 @@ additional_inputs=[
50
  ),
51
  gr.Slider(
52
  label="Temperature",
53
- value=0.9,
54
  minimum=0.0,
55
  maximum=1.0,
56
  step=0.05,
@@ -59,16 +59,16 @@ additional_inputs=[
59
  ),
60
  gr.Slider(
61
  label="Max new tokens",
62
- value=256,
63
  minimum=0,
64
- maximum=1048,
65
  step=64,
66
  interactive=True,
67
  info="The maximum numbers of new tokens",
68
  ),
69
  gr.Slider(
70
  label="Top-p (nucleus sampling)",
71
- value=0.90,
72
  minimum=0.0,
73
  maximum=1,
74
  step=0.05,
@@ -77,7 +77,7 @@ additional_inputs=[
77
  ),
78
  gr.Slider(
79
  label="Repetition penalty",
80
- value=1.2,
81
  minimum=1.0,
82
  maximum=2.0,
83
  step=0.05,
@@ -100,5 +100,5 @@ gr.ChatInterface(
100
  additional_inputs=additional_inputs,
101
  title="Dolphin 2.8 Gemma 2b",
102
  examples=examples,
103
- concurrency_limit=20,
104
  ).launch(show_api=False)
 
14
  return prompt
15
 
16
  def generate(
17
+ prompt, history, system_prompt, temperature=0.8, max_new_tokens=128, top_p=0.95, repetition_penalty=1.0,
18
  ):
19
  temperature = float(temperature)
20
  if temperature < 1e-2:
 
50
  ),
51
  gr.Slider(
52
  label="Temperature",
53
+ value=0.8,
54
  minimum=0.0,
55
  maximum=1.0,
56
  step=0.05,
 
59
  ),
60
  gr.Slider(
61
  label="Max new tokens",
62
+ value=128,
63
  minimum=0,
64
+ maximum=250,
65
  step=64,
66
  interactive=True,
67
  info="The maximum numbers of new tokens",
68
  ),
69
  gr.Slider(
70
  label="Top-p (nucleus sampling)",
71
+ value=0.95,
72
  minimum=0.0,
73
  maximum=1,
74
  step=0.05,
 
77
  ),
78
  gr.Slider(
79
  label="Repetition penalty",
80
+ value=1.0,
81
  minimum=1.0,
82
  maximum=2.0,
83
  step=0.05,
 
100
  additional_inputs=additional_inputs,
101
  title="Dolphin 2.8 Gemma 2b",
102
  examples=examples,
103
+ concurrency_limit=5,
104
  ).launch(show_api=False)