Spaces:
Running
Running
jhj0517
commited on
Commit
·
c7b95eb
1
Parent(s):
59921c3
fix spaces bug
Browse files
modules/whisper/faster_whisper_inference.py
CHANGED
|
@@ -9,7 +9,6 @@ import ctranslate2
|
|
| 9 |
import whisper
|
| 10 |
import gradio as gr
|
| 11 |
from argparse import Namespace
|
| 12 |
-
import spaces
|
| 13 |
|
| 14 |
from modules.whisper.whisper_parameter import *
|
| 15 |
from modules.whisper.whisper_base import WhisperBase
|
|
@@ -141,7 +140,6 @@ class FasterWhisperInference(WhisperBase):
|
|
| 141 |
model_paths[model_name] = os.path.join(webui_dir, self.model_dir, model_name)
|
| 142 |
return model_paths
|
| 143 |
|
| 144 |
-
@spaces.GPU
|
| 145 |
def get_available_compute_type(self):
|
| 146 |
if self.device == "cuda":
|
| 147 |
return ['float32', 'int8_float16', 'float16', 'int8', 'int8_float32']
|
|
|
|
| 9 |
import whisper
|
| 10 |
import gradio as gr
|
| 11 |
from argparse import Namespace
|
|
|
|
| 12 |
|
| 13 |
from modules.whisper.whisper_parameter import *
|
| 14 |
from modules.whisper.whisper_base import WhisperBase
|
|
|
|
| 140 |
model_paths[model_name] = os.path.join(webui_dir, self.model_dir, model_name)
|
| 141 |
return model_paths
|
| 142 |
|
|
|
|
| 143 |
def get_available_compute_type(self):
|
| 144 |
if self.device == "cuda":
|
| 145 |
return ['float32', 'int8_float16', 'float16', 'int8', 'int8_float32']
|