Hugging Face's logo

Spaces:
BSC-LT
/
asr-inference
Running on Zero

App Files Community
ankush13r commited on
Commit
94b8e1f
·
verified ·
1 Parent(s): e6b6783

Update whisper2.py

Browse files
Files changed (1) hide show
  1. whisper2.py +0 -1
whisper2.py CHANGED
@@ -7,7 +7,6 @@ MODEL_NAME = "openai/whisper-large-v3"
7
 
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
- device = "cpu"
11
 
12
  print("[ INFO ] Device: ", device)
13
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
7
 
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
10
 
11
  print("[ INFO ] Device: ", device)
12
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32