Spaces:
Runtime error
Runtime error
Samuel Schmidt
commited on
Commit
·
b836cf0
1
Parent(s):
7db538c
Add: Logging statement
Browse files- src/CLIP.py +1 -0
src/CLIP.py
CHANGED
@@ -18,6 +18,7 @@ class CLIPImageEncoder:
|
|
18 |
images = batch["image"]
|
19 |
print(images)
|
20 |
input = self.processor(images=images, return_tensors="pt")
|
|
|
21 |
with torch.no_grad():
|
22 |
image_features = self.model.get_image_features(**input)
|
23 |
#image_features = self.model(**input).last_hidden_state[:,0].cpu()
|
|
|
18 |
images = batch["image"]
|
19 |
print(images)
|
20 |
input = self.processor(images=images, return_tensors="pt")
|
21 |
+
print(input)
|
22 |
with torch.no_grad():
|
23 |
image_features = self.model.get_image_features(**input)
|
24 |
#image_features = self.model(**input).last_hidden_state[:,0].cpu()
|