Test: ran the code and checked for ffmpeg functionality

This commit is contained in:
Orihara Izaya
2023-10-09 18:13:32 +02:00
parent dc2ee55c4a
commit df1a3c075b
5 changed files with 11 additions and 5 deletions
+2 -4
View File
@@ -102,14 +102,12 @@ class Transcriber:
logger.exception("Failed to save transcript: %s", e)
raise
def _clear_memory(self):
@staticmethod
def _clear_memory():
"""
Clear GPU memory.
"""
try:
del self.transcription_model
del self.alignment_model
del self.metadata
gc.collect()
torch.cuda.empty_cache()
except Exception as e: