Spaces:
Running
Running
Test params
Browse files- translate.py +2 -2
translate.py
CHANGED
|
@@ -71,8 +71,8 @@ def main(
|
|
| 71 |
|
| 72 |
accelerator = Accelerator(
|
| 73 |
mixed_precision=precision if precision != "32" else "no",
|
| 74 |
-
split_batches=False,
|
| 75 |
-
dispatch_batches=True,
|
| 76 |
)
|
| 77 |
|
| 78 |
print(f"Loading tokenizer {model_name}...")
|
|
|
|
| 71 |
|
| 72 |
accelerator = Accelerator(
|
| 73 |
mixed_precision=precision if precision != "32" else "no",
|
| 74 |
+
#split_batches=False,
|
| 75 |
+
#dispatch_batches=True,
|
| 76 |
)
|
| 77 |
|
| 78 |
print(f"Loading tokenizer {model_name}...")
|