Spaces:
Running
Running
[Fix Bug] Max length param was ignored
Browse files- 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 |
)
|