Spaces:

shivammehta25
/

Diff-TTSG

Runtime error

Shivam Mehta commited on Jun 27, 2023

Commit

bb50f7a

1 Parent(s): 9d6551a

cuda fixes

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,7 @@ def load_vocoder(checkpoint_path):
 cmu = cmudict.CMUDict(CMU_DICT_PATH)
 def process_text(text: str):
     x = torch.LongTensor(intersperse(text_to_sequence(text, dictionary=cmu), len(symbols))).to(device)[None]
-    x_lengths = torch.LongTensor([x.shape[-1]]).cuda()
     x_phones = sequence_to_text(x.squeeze(0).tolist())
     return {
         'x_orig': text,

 cmu = cmudict.CMUDict(CMU_DICT_PATH)
 def process_text(text: str):
     x = torch.LongTensor(intersperse(text_to_sequence(text, dictionary=cmu), len(symbols))).to(device)[None]
+    x_lengths = torch.LongTensor([x.shape[-1]]).to(device)
     x_phones = sequence_to_text(x.squeeze(0).tolist())
     return {
         'x_orig': text,