VALL-E-X

Build error

Plachta commited on Aug 28, 2023

Commit

515113a

•

1 Parent(s): e0708dc

Fix OOM

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,8 +44,8 @@ text_tokenizer = PhonemeBpeTokenizer(tokenizer_path="./utils/g2p/bpe_69.json")
 text_collater = get_text_token_collater()
 device = torch.device("cpu")
-# if torch.cuda.is_available():
-#     device = torch.device("cuda", 0)
 # VALL-E-X model
 model = VALLE(

 text_collater = get_text_token_collater()
 device = torch.device("cpu")
+if torch.cuda.is_available():
+    device = torch.device("cuda", 0)
 # VALL-E-X model
 model = VALLE(