Skip to content

GitLab

Commit d18d47be authored by Lysandre's avatar Lysandre
Browse files

run_generation style

parent ff6f1492
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/run_generation.py
View file @ d18d47be
...@@ -221,7 +221,7 @@ def main(): ...@@ -221,7 +221,7 @@ def main():
top_k=args.k, top_k=args.k,
top_p=args.p, top_p=args.p,
repetition_penalty=args.repetition_penalty, repetition_penalty=args.repetition_penalty,
do_sample=True do_sample=True,
) )
# Batch size == 1. to add more examples please use num_return_sequences > 1 # Batch size == 1. to add more examples please use num_return_sequences > 1
......
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