jhj0517 commited on
Commit
81509c3
·
1 Parent(s): d868316

fix wrong device

Browse files
Files changed (1) hide show
  1. modules/whisper_Inference.py +0 -1
modules/whisper_Inference.py CHANGED
@@ -29,7 +29,6 @@ class WhisperInference(BaseInterface):
29
  self.device = "mps"
30
  else:
31
  self.device = "cpu"
32
- self.device = "cuda" if torch.cuda.is_available() else "cpu"
33
  self.available_compute_types = ["float16", "float32"]
34
  self.current_compute_type = "float16" if self.device == "cuda" else "float32"
35
  self.model_dir = os.path.join("models", "Whisper")
 
29
  self.device = "mps"
30
  else:
31
  self.device = "cpu"
 
32
  self.available_compute_types = ["float16", "float32"]
33
  self.current_compute_type = "float16" if self.device == "cuda" else "float32"
34
  self.model_dir = os.path.join("models", "Whisper")