Spaces:

jesseplusplus
/

easy-translate

Running

Iker commited on Jul 1, 2022

Commit

56e0241

•

1 Parent(s): ea68187

Fix bug

Files changed (1) hide show

translate.py CHANGED Viewed

@@ -61,8 +61,8 @@ def main(
     num_beams: int = 4,
 ):
-    if not os.path.exists(os.path.dirname(output_path)):
-        os.makedirs(os.path.dirname(output_path))
     accelerator = Accelerator(
         mixed_precision=precision if precision != "32" else "no", split_batches=True

     num_beams: int = 4,
 ):
+    if not os.path.exists(os.path.abspath(os.path.dirname(output_path))):
+        os.makedirs(os.path.abspath(os.path.dirname(output_path)))
     accelerator = Accelerator(
         mixed_precision=precision if precision != "32" else "no", split_batches=True