Update README.md
Browse files
README.md
CHANGED
@@ -152,7 +152,7 @@ probs = logits_per_image.softmax(dim=1) # we can take the softmax to get the lab
|
|
152 |
print("Label probabilities:", probs)
|
153 |
print("Logits per image:", logits_per_image)
|
154 |
|
155 |
-
# Compute image and text embeddings
|
156 |
|
157 |
text_features = model.get_text_features(inputs["input_ids"], inputs["attention_mask"])
|
158 |
image_features = model.get_image_features(inputs["pixel_values"])
|
|
|
152 |
print("Label probabilities:", probs)
|
153 |
print("Logits per image:", logits_per_image)
|
154 |
|
155 |
+
# Compute image and text embeddings separately
|
156 |
|
157 |
text_features = model.get_text_features(inputs["input_ids"], inputs["attention_mask"])
|
158 |
image_features = model.get_image_features(inputs["pixel_values"])
|