Skip to content

GitLab

Unverified Commit 46078e1b authored by Thomas Wolf's avatar Thomas Wolf Committed by GitHub
Browse files

Merge pull request #496 from 8enmann/patch-1 

[run_gpt2.py] temperature should be a float, not int
parents b8686130 87677fcc
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/run_gpt2.py
View file @ 46078e1b
...@@ -58,7 +58,7 @@ def run_model(): ...@@ -58,7 +58,7 @@ def run_model():
parser.add_argument("--nsamples", type=int, default=1) parser.add_argument("--nsamples", type=int, default=1)
parser.add_argument("--batch_size", type=int, default=-1) parser.add_argument("--batch_size", type=int, default=-1)
parser.add_argument("--length", type=int, default=-1) parser.add_argument("--length", type=int, default=-1)
parser.add_argument("--temperature", type=int, default=1) parser.add_argument("--temperature", type=float, default=1.0)
parser.add_argument("--top_k", type=int, default=0) parser.add_argument("--top_k", type=int, default=0)
parser.add_argument('--unconditional', action='store_true', help='If true, unconditional generation.') parser.add_argument('--unconditional', action='store_true', help='If true, unconditional generation.')
args = parser.parse_args() args = parser.parse_args()
......
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