masanorihirano commited on
Commit
6fbdb04
·
1 Parent(s): 7935419
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -194,6 +194,7 @@ def evaluate(
194
  )
195
  input_ids = inputs["input_ids"].to(device)
196
  generation_config = GenerationConfig(
 
197
  temperature=temperature,
198
  top_p=top_p,
199
  top_k=top_k,
@@ -258,7 +259,6 @@ description = (
258
  "It is trained on the [izumi-lab/llm-japanese-dataset](https://huggingface.co/datasets/izumi-lab/llm-japanese-dataset) dataset. "
259
  "For more information, please visit [the project's website](https://llm.msuzuki.me). "
260
  "This model can output up to 256 tokens, but the maximum number of tokens is 227 due to the GPU memory limit of HuggingFace Space. "
261
- "Currently temperature is ineffective because num_beam is set to 1 due to lack of GPU memory. "
262
  "It takes about **1 minute** to output. When access is concentrated, the operation may become slow."
263
  )
264
  with gr.Blocks(
@@ -289,7 +289,7 @@ with gr.Blocks(
289
  value=0.7,
290
  step=0.05,
291
  interactive=True,
292
- label="Temperature (Currently ineffective due to insufficient GPU memory)",
293
  )
294
  max_tokens = gr.Slider(
295
  minimum=20,
 
194
  )
195
  input_ids = inputs["input_ids"].to(device)
196
  generation_config = GenerationConfig(
197
+ do_sample=True,
198
  temperature=temperature,
199
  top_p=top_p,
200
  top_k=top_k,
 
259
  "It is trained on the [izumi-lab/llm-japanese-dataset](https://huggingface.co/datasets/izumi-lab/llm-japanese-dataset) dataset. "
260
  "For more information, please visit [the project's website](https://llm.msuzuki.me). "
261
  "This model can output up to 256 tokens, but the maximum number of tokens is 227 due to the GPU memory limit of HuggingFace Space. "
 
262
  "It takes about **1 minute** to output. When access is concentrated, the operation may become slow."
263
  )
264
  with gr.Blocks(
 
289
  value=0.7,
290
  step=0.05,
291
  interactive=True,
292
+ label="Temperature",
293
  )
294
  max_tokens = gr.Slider(
295
  minimum=20,