Spaces:
Running
Running
jhj0517
commited on
Commit
·
87cbb02
1
Parent(s):
b39eb49
Use float32 for cpu
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -375,6 +375,8 @@ class WhisperBase(ABC):
|
|
375 |
def get_compute_type(self):
|
376 |
if "float16" in self.available_compute_types:
|
377 |
return "float16"
|
|
|
|
|
378 |
else:
|
379 |
return self.available_compute_types[0]
|
380 |
|
|
|
375 |
def get_compute_type(self):
|
376 |
if "float16" in self.available_compute_types:
|
377 |
return "float16"
|
378 |
+
if "float32" in self.available_compute_types:
|
379 |
+
return "float32"
|
380 |
else:
|
381 |
return self.available_compute_types[0]
|
382 |
|