LAP-DEV commited on
Commit
eb96547
·
verified ·
1 Parent(s): 8e42e26

Update modules/whisper/whisper_base.py

Browse files
Files changed (1) hide show
  1. modules/whisper/whisper_base.py +8 -8
modules/whisper/whisper_base.py CHANGED
@@ -568,14 +568,14 @@ class WhisperBase(ABC):
568
  write_file(content, output_path)
569
  return content, output_path
570
 
571
- def offload(self):
572
- """Offload the model and free up the memory"""
573
- if self.model is not None:
574
- del self.model
575
- self.model = None
576
- if self.device == "cuda":
577
- self.release_cuda_memory()
578
- gc.collect()
579
 
580
  @staticmethod
581
  def format_time(elapsed_time: float) -> str:
 
568
  write_file(content, output_path)
569
  return content, output_path
570
 
571
+ # def offload(self):
572
+ # """Offload the model and free up the memory"""
573
+ # if self.model is not None:
574
+ # del self.model
575
+ # self.model = None
576
+ # if self.device == "cuda":
577
+ # self.release_cuda_memory()
578
+ # gc.collect()
579
 
580
  @staticmethod
581
  def format_time(elapsed_time: float) -> str: