Spaces:
Running
Running
jhj0517
commited on
Commit
·
81509c3
1
Parent(s):
d868316
fix wrong device
Browse files
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")
|