Spaces:
Runtime error
Runtime error
jhj0517 commited on
Commit ·
3576b73
1
Parent(s): 2205c35
debug release function
Browse files
modules/whisper/whisper_base.py
CHANGED
|
@@ -404,8 +404,8 @@ class WhisperBase(ABC):
|
|
| 404 |
|
| 405 |
return time_str.strip()
|
| 406 |
|
| 407 |
-
@
|
| 408 |
-
def release_cuda_memory(
|
| 409 |
if torch.cuda.is_available():
|
| 410 |
torch.cuda.empty_cache()
|
| 411 |
torch.cuda.reset_max_memory_allocated()
|
|
|
|
| 404 |
|
| 405 |
return time_str.strip()
|
| 406 |
|
| 407 |
+
@staticmethod
|
| 408 |
+
def release_cuda_memory():
|
| 409 |
if torch.cuda.is_available():
|
| 410 |
torch.cuda.empty_cache()
|
| 411 |
torch.cuda.reset_max_memory_allocated()
|