Spaces:
Running
Running
jhj0517
commited on
Commit
·
4ee4dc9
1
Parent(s):
e874115
debug decoration
Browse files
modules/whisper/faster_whisper_inference.py
CHANGED
|
@@ -153,6 +153,7 @@ class FasterWhisperInference(WhisperBase):
|
|
| 153 |
return ['float32', 'int8_float16', 'float16', 'int8', 'int8_float32']
|
| 154 |
return ['int16', 'float32', 'int8', 'int8_float32']
|
| 155 |
|
|
|
|
| 156 |
def get_device(self):
|
| 157 |
# To fix huggingface spaces bug just return cuda
|
| 158 |
return "cuda"
|
|
|
|
| 153 |
return ['float32', 'int8_float16', 'float16', 'int8', 'int8_float32']
|
| 154 |
return ['int16', 'float32', 'int8', 'int8_float32']
|
| 155 |
|
| 156 |
+
@spaces.GPU
|
| 157 |
def get_device(self):
|
| 158 |
# To fix huggingface spaces bug just return cuda
|
| 159 |
return "cuda"
|