sts

Running on A100

vineelpratap commited on Sep 26, 2023

Commit

1b7f5da

•

1 Parent(s): 6f53272

Update asr.py

Files changed (1) hide show

asr.py CHANGED Viewed

@@ -107,7 +107,7 @@ def transcribe(
     with torch.no_grad():
         outputs = model(**inputs).logits
-    if lang_code != "eng" and False:
         ids = torch.argmax(outputs, dim=-1)[0]
         transcription = processor.decode(ids)
     else:

     with torch.no_grad():
         outputs = model(**inputs).logits
+    if lang_code != "eng" or True:
         ids = torch.argmax(outputs, dim=-1)[0]
         transcription = processor.decode(ids)
     else: