core/config.py

import torch

device = 0 if torch.cuda.is_available() else -1
cache_dir = "./models_cache"