Skip to content

Commit 058dc2b

Browse files
handle MPS device in emissions
Co-authored-by: Mahmoud Ashraf <[email protected]>
1 parent 134666e commit 058dc2b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ctc_forced_aligner/alignment_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ def get_alignments(
237237

238238
blank_id = dictionary.get("<blank>", tokenizer.pad_token_id)
239239

240-
if emissions.is_cuda:
240+
if not emissions.is_cpu:
241241
emissions = emissions.cpu()
242242
targets = np.asarray([token_indices], dtype=np.int64)
243243

0 commit comments

Comments
 (0)