marker-io / marker /utils.py
Ritvik19's picture
Add all files and directories
c8a32e7
raw
history blame contribute delete
156 Bytes
import torch
from marker.settings import settings
def flush_cuda_memory():
if settings.TORCH_DEVICE_MODEL == "cuda":
torch.cuda.empty_cache()