vilarin commited on
Commit
d992f1a
·
verified ·
1 Parent(s): ed629f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -50,8 +50,8 @@ model = OlmoeForCausalLM.from_pretrained(
50
  def stream_chat(
51
  message: str,
52
  history: list,
53
- temperature: float = 0.3,
54
- max_new_tokens: int = 1024,
55
  top_p: float = 1.0,
56
  top_k: int = 20,
57
  penalty: float = 1.2,
@@ -110,15 +110,15 @@ with gr.Blocks(css=CSS, theme="Nymbo/Nymbo_Theme") as demo:
110
  minimum=0,
111
  maximum=1,
112
  step=0.1,
113
- value=0.3,
114
  label="Temperature",
115
  render=False,
116
  ),
117
  gr.Slider(
118
  minimum=128,
119
- maximum=8192,
120
  step=1,
121
- value=1024,
122
  label="Max new tokens",
123
  render=False,
124
  ),
 
50
  def stream_chat(
51
  message: str,
52
  history: list,
53
+ temperature: float = 0.5,
54
+ max_new_tokens: int = 256,
55
  top_p: float = 1.0,
56
  top_k: int = 20,
57
  penalty: float = 1.2,
 
110
  minimum=0,
111
  maximum=1,
112
  step=0.1,
113
+ value=0.5,
114
  label="Temperature",
115
  render=False,
116
  ),
117
  gr.Slider(
118
  minimum=128,
119
+ maximum=4096,
120
  step=1,
121
+ value=256,
122
  label="Max new tokens",
123
  render=False,
124
  ),