Spaces:

jesseplusplus
/

easy-translate

Running

Iker commited on Sep 1, 2022

Commit

1cb8a41

1 Parent(s): 7f820ab

Test params

Files changed (1) hide show

translate.py CHANGED Viewed

@@ -71,8 +71,8 @@ def main(
     accelerator = Accelerator(
         mixed_precision=precision if precision != "32" else "no",
-        # split_batches=False,
-        # dispatch_batches=False,
     )
     print(f"Loading tokenizer {model_name}...")

     accelerator = Accelerator(
         mixed_precision=precision if precision != "32" else "no",
+        split_batches=False,
+        dispatch_batches=False,
     )
     print(f"Loading tokenizer {model_name}...")