huy-ha commited on
Commit
4181a28
·
1 Parent(s): af7b806

fix log device

Browse files
Files changed (1) hide show
  1. CLIP/clip/__init__.py +2 -5
CLIP/clip/__init__.py CHANGED
@@ -69,11 +69,8 @@ class ClipWrapper:
69
  @classmethod
70
  def check_initialized(cls, clip_model_type="ViT-B/32", **kwargs):
71
  if cls.clip_gradcam is None:
72
- ClipWrapper(
73
- clip_model_type=clip_model_type,
74
- device="cuda" if torch.cuda.is_available() else "cpu",
75
- **kwargs
76
- )
77
  print("using", device)
78
 
79
  @classmethod
 
69
  @classmethod
70
  def check_initialized(cls, clip_model_type="ViT-B/32", **kwargs):
71
  if cls.clip_gradcam is None:
72
+ device = "cuda" if torch.cuda.is_available() else "cpu"
73
+ ClipWrapper(clip_model_type=clip_model_type, device=device, **kwargs)
 
 
 
74
  print("using", device)
75
 
76
  @classmethod