huy-ha commited on
Commit
af7b806
·
1 Parent(s): 8fbac9e

log device

Browse files
Files changed (1) hide show
  1. CLIP/clip/__init__.py +1 -0
CLIP/clip/__init__.py CHANGED
@@ -74,6 +74,7 @@ class ClipWrapper:
74
  device="cuda" if torch.cuda.is_available() else "cpu",
75
  **kwargs
76
  )
 
77
 
78
  @classmethod
79
  def get_clip_text_feature(cls, string):
 
74
  device="cuda" if torch.cuda.is_available() else "cpu",
75
  **kwargs
76
  )
77
+ print("using", device)
78
 
79
  @classmethod
80
  def get_clip_text_feature(cls, string):