tts-generation-webui/tts_webui/utils/torch_clear_memory.py

import torch
import gc


def torch_clear_memory():
    try:
        if torch.cuda.is_available():
            torch.cuda.empty_cache()
    except Exception:
        pass
    gc.collect()