SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

run_generation style

Browse Source
This commit is contained in:
Lysandre
2020-01-31 12:05:48 -05:00
parent ff6f1492e8
commit d18d47be67
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/run_generation.py
View File

@@ -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