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