Unverified Commit 45b0b1ff authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[s2s] fix kwarg typo (#7196)

parent 0203ad43
...@@ -152,7 +152,7 @@ def run_generate(): ...@@ -152,7 +152,7 @@ def run_generate():
json_save_dir, json_save_dir,
args.model_name, args.model_name,
type_path=args.type_path, type_path=args.type_path,
batch_size=args.bs, bs=args.bs,
fp16=args.fp16, fp16=args.fp16,
task=args.task, task=args.task,
local_rank=args.local_rank, local_rank=args.local_rank,
......
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