Vikhrmodels
/

Borealis

Automatic Speech Recognition

text-generation

Model card Files Files and versions

Anonumous commited on Sep 11, 2025

Commit

6791bf0

·

verified ·

1 Parent(s): ff3cbdb

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ proc = extractor(
         return_tensors="pt",
     )
-mel = proc.input_features.squeeze(0).to(device)
 att_mask = proc.attention_mask.squeeze(0).to("cuda")
 with torch.inference_mode():

         return_tensors="pt",
     )
+mel = proc.input_features.squeeze(0).to("cuda")
 att_mask = proc.attention_mask.squeeze(0).to("cuda")
 with torch.inference_mode():