Spaces:
Running
Running
jhj0517
commited on
Commit
·
564cf04
1
Parent(s):
342afe7
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']
|