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 |
)
|