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 |
|