Spaces:
Running
Running
jhj0517
commited on
Commit
·
59921c3
1
Parent(s):
24956db
fix spaces bug
Browse files
modules/whisper/faster_whisper_inference.py
CHANGED
|
@@ -144,8 +144,8 @@ class FasterWhisperInference(WhisperBase):
|
|
| 144 |
@spaces.GPU
|
| 145 |
def get_available_compute_type(self):
|
| 146 |
if self.device == "cuda":
|
| 147 |
-
return
|
| 148 |
-
return
|
| 149 |
|
| 150 |
@staticmethod
|
| 151 |
def get_device():
|
|
|
|
| 144 |
@spaces.GPU
|
| 145 |
def get_available_compute_type(self):
|
| 146 |
if self.device == "cuda":
|
| 147 |
+
return ['float32', 'int8_float16', 'float16', 'int8', 'int8_float32']
|
| 148 |
+
return ['int16', 'float32', 'int8', 'int8_float32']
|
| 149 |
|
| 150 |
@staticmethod
|
| 151 |
def get_device():
|