do_sample should be set to True in run_generation.py
This commit is contained in:
@@ -221,6 +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
|
||||||
)
|
)
|
||||||
|
|
||||||
# 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
|
||||||
|
|||||||
Reference in New Issue
Block a user