Unverified Commit 03e46c1d authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[s2s] fix kwargs style (#7488)

parent 6fe8a693
...@@ -176,7 +176,8 @@ def run_generate(): ...@@ -176,7 +176,8 @@ def run_generate():
max_source_length=args.max_source_length, max_source_length=args.max_source_length,
num_return_sequences=args.num_return_sequences, num_return_sequences=args.num_return_sequences,
prefix=args.prefix, prefix=args.prefix,
dataset_kwargs=dataset_kwargs ** generate_kwargs, dataset_kwargs=dataset_kwargs,
**generate_kwargs,
) )
if args.local_rank <= 0: if args.local_rank <= 0:
......
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