config/settings.py

import torch

MODEL_ID = "openai/clip-vit-base-patch32"

TORCH_DTYPE = torch.bfloat16
ATTN_IMPLEMENTATION = "sdpa"

DEVICE = "cuda" if torch.cuda.is_available() else "cpu"