federicocosta1989 commited on
Commit
3f0edbb
·
verified ·
1 Parent(s): cd50548

Update whisper_cs_dev.py

Browse files

Removed the HACK from get_settings

Files changed (1) hide show
  1. whisper_cs_dev.py +1 -2
whisper_cs_dev.py CHANGED
@@ -12,8 +12,7 @@ def get_settings():
12
 
13
  if DEBUG_MODE: print(f"Entering get_settings function...")
14
 
15
- # HACK hardcoding this to try
16
- is_cuda_available = True #torch.cuda.is_available()
17
  if is_cuda_available:
18
  device = "cuda"
19
  compute_type = "float16"
 
12
 
13
  if DEBUG_MODE: print(f"Entering get_settings function...")
14
 
15
+ is_cuda_available = torch.cuda.is_available()
 
16
  if is_cuda_available:
17
  device = "cuda"
18
  compute_type = "float16"