c8a32e7
1
2
3
4
5
6
7
8
import torch from marker.settings import settings def flush_cuda_memory(): if settings.TORCH_DEVICE_MODEL == "cuda": torch.cuda.empty_cache()
