jhj0517 commited on
Commit
3576b73
·
1 Parent(s): 2205c35

debug release function

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +2 -2
modules/whisper/whisper_base.py CHANGED
@@ -404,8 +404,8 @@ class WhisperBase(ABC):
404
 
405
  return time_str.strip()
406
 
407
- @spaces.GPU(duration=120)
408
- def release_cuda_memory(self):
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()