Update bert_gen.py

2023-09-05 20:27:00 +08:00
parent 97ec85a510
commit c7b796df77
1 changed files with 1 additions and 1 deletions
--- a/bert_gen.py
+++ b/bert_gen.py
@@ -14,7 +14,7 @@ def process_line(line):
    if torch.cuda.is_available():
        gpu_id = rank % torch.cuda.device_count()
        device = torch.device(f"cuda:{gpu_id}")
-    logger.info(f"Rank {rank} uses device {device}")
+    #logger.info(f"Rank {rank} uses device {device}")
    wav_path, _, language_str, text, phones, tone, word2ph = line.strip().split("|")
    phone = phones.split(" ")
    tone = [int(i) for i in tone.split(" ")]