Update README.md
Browse files
README.md
CHANGED
|
@@ -132,8 +132,6 @@ inputs_img = process_messages(messages_img)
|
|
| 132 |
with torch.no_grad():
|
| 133 |
embeddings_txt = model(**inputs_txt) # [1, 1536]
|
| 134 |
embeddings_img = model(**inputs_img) # [1, 1536]
|
| 135 |
-
print(f"embeddings_txt.shape: {embeddings_txt.shape}")
|
| 136 |
-
print(f"embeddings_img.shape: {embeddings_img.shape}")
|
| 137 |
|
| 138 |
print(torch.matmul(embeddings_txt, embeddings_img.T))
|
| 139 |
# tensor([[0.7266]], dtype=torch.bfloat16)
|
|
|
|
| 132 |
with torch.no_grad():
|
| 133 |
embeddings_txt = model(**inputs_txt) # [1, 1536]
|
| 134 |
embeddings_img = model(**inputs_img) # [1, 1536]
|
|
|
|
|
|
|
| 135 |
|
| 136 |
print(torch.matmul(embeddings_txt, embeddings_img.T))
|
| 137 |
# tensor([[0.7266]], dtype=torch.bfloat16)
|