Views
No views yet
1 2import torch 3import clip 4 5device = "cuda" if torch.cuda.is_available() else "cpu" 6model, preprocess = clip.load("ViT-B/32", device=device) 7model.load_state_dict(torch.load("negationclip_ViT-B32.pth", map_location=device))