Update nemo_align.py

nemo_align.py

@@ -442,13 +442,11 @@ def align_tdt_to_ctc_timestamps(tdt_txt, model, audio_filepath):
         raise ValueError("Currently supporting hybrid models")
     
     if torch.cuda.is_available():
-        enable = True
         viterbi_device = torch.device('cuda')
     else:
-        enable = False
         viterbi_device = torch.device('cpu')
     
-    with torch.cuda.amp.autocast(enabled=enable, dtype=torch.bfloat16):
+    with torch.cuda.amp.autocast(enabled=False, dtype=torch.bfloat16):
         with torch.inference_mode():
             hypotheses = model.transcribe([audio_filepath], return_hypotheses=True, batch_size=1)