From c6b6874b34bda8fd1be3991b64b9665d697f5272 Mon Sep 17 00:00:00 2001
From: Shengqiang Li <49022799+Shengqiang-Li@users.noreply.github.com>
Date: Tue, 5 Mar 2024 21:10:04 +0800
Subject: [PATCH] [vits] fix the bug in train.py (#211)

---
 wetts/vits/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wetts/vits/train.py b/wetts/vits/train.py
index 29e1e73..d59a81a 100644
--- a/wetts/vits/train.py
+++ b/wetts/vits/train.py
@@ -35,7 +35,7 @@ def main():
     world_size = int(os.environ.get('WORLD_SIZE', 1))
     local_rank = int(os.environ.get('LOCAL_RANK', 0))
     rank = int(os.environ.get('RANK', 0))
-    torch.torch.cuda.set_device(local_rank)
+    torch.cuda.set_device(local_rank)
     dist.init_process_group("nccl")
     if rank == 0:
         logger = task.get_logger(hps.model_dir)