CLIP on GPU

app.py

@@ -51,7 +51,7 @@ model = get_model('cc12m_1_cfg')()
 _, side_y, side_x = model.shape
 model.load_state_dict(torch.load(cc12m_model, map_location='cpu'))
 model = model.half().cuda().eval().requires_grad_(False)
-clip_model = clip.load(model.clip_model, jit=False, device='cpu')[0]
+clip_model = clip.load(model.clip_model, jit=False, device='cuda')[0]
 clip_model.eval().requires_grad_(False)
 normalize = transforms.Normalize(mean=[0.48145466, 0.4578275, 0.40821073],
                                      std=[0.26862954, 0.26130258, 0.27577711])