Spaces:
Running
Running
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()
|