Iker commited on
Commit
9f0798a
·
1 Parent(s): 02502dd

[Fix Bug] Max length param was ignored

Browse files
Files changed (1) hide show
  1. translate.py +1 -0
translate.py CHANGED
@@ -255,6 +255,7 @@ if __name__ == "__main__":
255
  starting_batch_size=args.starting_batch_size,
256
  model_name=args.model_name,
257
  cache_dir=args.cache_dir,
 
258
  num_beams=args.num_beams,
259
  precision=args.precision,
260
  )
 
255
  starting_batch_size=args.starting_batch_size,
256
  model_name=args.model_name,
257
  cache_dir=args.cache_dir,
258
+ max_length=args.max_length,
259
  num_beams=args.num_beams,
260
  precision=args.precision,
261
  )