Spaces:
Running
Running
jhj0517
commited on
Commit
·
d79b9a9
1
Parent(s):
be6b9ac
fix spaces bug
Browse files
modules/whisper/faster_whisper_inference.py
CHANGED
@@ -142,7 +142,7 @@ class FasterWhisperInference(WhisperBase):
|
|
142 |
model_paths[model_name] = os.path.join(webui_dir, self.model_dir, model_name)
|
143 |
return model_paths
|
144 |
|
145 |
-
@spaces.
|
146 |
def get_available_compute_type(self):
|
147 |
if self.device == "cuda":
|
148 |
return ctranslate2.get_supported_compute_types("cuda")
|
|
|
142 |
model_paths[model_name] = os.path.join(webui_dir, self.model_dir, model_name)
|
143 |
return model_paths
|
144 |
|
145 |
+
@spaces.GPU
|
146 |
def get_available_compute_type(self):
|
147 |
if self.device == "cuda":
|
148 |
return ctranslate2.get_supported_compute_types("cuda")
|