federicocosta1989 commited on
Commit
f8893d8
·
verified ·
1 Parent(s): a8f670e

Update whisper_cs.py

Browse files
Files changed (1) hide show
  1. whisper_cs.py +2 -2
whisper_cs.py CHANGED
@@ -14,8 +14,8 @@ torch_dtype = torch.float32
14
  DEBUG_MODE = True
15
  MODEL_PATH_V2 = "langtech-veu/whisper-timestamped-cs"
16
  MODEL_PATH_V2_FAST = "langtech-veu/faster-whisper-timestamped-cs"
17
- DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
18
- print("[INFO] CUDA available:", torch.cuda.is_available())
19
 
20
  def clean_text(input_text):
21
  remove_chars = ['.', ',', ';', ':', '¿', '?', '«', '»', '-', '¡', '!', '@',
 
14
  DEBUG_MODE = True
15
  MODEL_PATH_V2 = "langtech-veu/whisper-timestamped-cs"
16
  MODEL_PATH_V2_FAST = "langtech-veu/faster-whisper-timestamped-cs"
17
+ #DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
18
+ #print("[INFO] CUDA available:", torch.cuda.is_available())
19
 
20
  def clean_text(input_text):
21
  remove_chars = ['.', ',', ';', ':', '¿', '?', '«', '»', '-', '¡', '!', '@',