Commit
·
6fbdb04
1
Parent(s):
7935419
update
Browse files
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
|
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,
|