fix: preventing multi-process

app.py

@@ -202,7 +202,8 @@ def download_expdir(exp_dir: str) -> str:
     return f"{exp_dir}.zip"
 
 
-def restore_expdir(zip: str, exp_dir: str) -> str:
+def restore_expdir(zip: str) -> str:
+    exp_dir = tempfile.mkdtemp()
     shutil.unpack_archive(zip, exp_dir)
     return exp_dir
 
@@ -290,7 +291,7 @@ with gr.Blocks() as app:
 
     restore_btn.click(
         fn=restore_expdir,
-        inputs=[restore_zip_file, exp_dir],
+        inputs=[restore_zip_file],
         outputs=[exp_dir],
     )
 

infer/modules/train/train.py

@@ -147,7 +147,7 @@ def run(rank, n_gpus, hps, logger: logging.Logger, state):
         collate_fn = TextAudioCollate()
     train_loader = DataLoader(
         train_dataset,
-        num_workers=4,
+        # num_workers=4,
         shuffle=False,
         pin_memory=True,
         collate_fn=collate_fn,