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()
|