Update README.md

README.md

@@ -130,10 +130,8 @@ inputs_txt = process_messages(messages_txt)
 inputs_img = process_messages(messages_img)
 
 with torch.no_grad():
-    embeddings_txt = model(**inputs_txt)  # [1, 1536]
-    embeddings_img = model(**inputs_img)  # [1, 1536]
-    print(f"embeddings_txt.shape: {embeddings_txt.shape}")
-    print(f"embeddings_img.shape: {embeddings_img.shape}")
+    embeddings_txt = model(**inputs_txt)  # [1, 3584]
+    embeddings_img = model(**inputs_img)  # [1, 3584]
 
     print(torch.matmul(embeddings_txt, embeddings_img.T))
     # tensor([[0.7578]], dtype=torch.bfloat16)
@@ -175,8 +173,8 @@ inputs_txt = process_messages(messages_txt)
 inputs_vid = process_messages(messages_vid)
 
 with torch.no_grad():
-    embeddings_txt = model(**inputs_txt)  # [1, 1536]
-    embeddings_vid = model(**inputs_vid)  # [1, 1536]
+    embeddings_txt = model(**inputs_txt)  # [1, 3584]
+    embeddings_vid = model(**inputs_vid)  # [1, 3584]
 
     print(torch.matmul(embeddings_txt, embeddings_vid.T))
     # tensor([[0.4883]], dtype=torch.bfloat16)
@@ -213,8 +211,8 @@ inputs_qry = process_messages(messages_qry)
 inputs_tgt = process_messages(messages_tgt)
 
 with torch.no_grad():
-    embeddings_qry = model(**inputs_qry)  # [1, 1536]
-    embeddings_tgt = model(**inputs_tgt)  # [1, 1536]
+    embeddings_qry = model(**inputs_qry)  # [1, 3584]
+    embeddings_tgt = model(**inputs_tgt)  # [1, 3584]
 
     print(torch.matmul(embeddings_qry, embeddings_tgt.T))
     # tensor([[0.6719]], dtype=torch.bfloat16)