Spaces:

HF-Pawan
/

Open-AI-Zero-Shot-Image-Classification

Running

File size: 171 Bytes

5aa6736

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"