fix bug in bert_gen.py (#54)

fishaudio · Oct 9, 2023 · dc48bb4 · dc48bb4
1 parent b57dfd7
commit dc48bb4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bert_gen.py b/bert_gen.py
@@ -14,6 +14,8 @@ def process_line(line):
     if torch.cuda.is_available():
         gpu_id = rank % torch.cuda.device_count()
         device = torch.device(f"cuda:{gpu_id}")
+    else:
+        device = torch.device("cpu")
     wav_path, _, language_str, text, phones, tone, word2ph = line.strip().split("|")
     phone = phones.split(" ")
     tone = [int(i) for i in tone.split(" ")]