Update modules/whisper/whisper_base.py
Browse files
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:
|