Spaces:
Sleeping
Sleeping
jhj0517
commited on
Commit
·
342afe7
1
Parent(s):
9212c90
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():
|