Skip to content

GitLab

Unverified Commit 88524526 authored by echo840's avatar echo840 Committed by GitHub
Browse files

Update demo.py

parent 9a8886d4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
demo.py
View file @ 88524526
......@@ -152,7 +152,7 @@ def _launch_demo(args, model, tokenizer):
num_beams=1,
max_new_tokens=512,
min_new_tokens=1,
length_penalty=3,
length_penalty=1,
num_return_sequences=1,
output_hidden_states=True,
use_cache=True,
......@@ -204,7 +204,7 @@ def _launch_demo(args, model, tokenizer):
temperature=0.7,
max_new_tokens=250,
min_new_tokens=1,
length_penalty=3,
length_penalty=1,
num_return_sequences=1,
output_hidden_states=True,
use_cache=True,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment