Spaces:
Build error
Build error
log device
Browse files- 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):
|