Skip to content

GitLab

Unverified Commit 87677fcc authored by Ben Mann's avatar Ben Mann Committed by GitHub
Browse files

[run_gpt2.py] temperature should be a float, not int

parent 3d78e226
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/run_gpt2.py
View file @ 87677fcc
...@@ -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