Update README.md
Browse files
README.md
CHANGED
|
@@ -146,7 +146,7 @@ mean_embeddings = sum_embeddings / expanded_mask.sum(dim=1)
|
|
| 146 |
|
| 147 |
# Output
|
| 148 |
print("EOS (Last Token) Embeddings:", eos_embeddings)
|
| 149 |
-
print("Mean Pooling Embeddings:", mean_embeddings
|
| 150 |
|
| 151 |
# ============================================================================
|
| 152 |
# Additional notes:
|
|
|
|
| 146 |
|
| 147 |
# Output
|
| 148 |
print("EOS (Last Token) Embeddings:", eos_embeddings)
|
| 149 |
+
print("Mean Pooling Embeddings:", mean_embeddings)
|
| 150 |
|
| 151 |
# ============================================================================
|
| 152 |
# Additional notes:
|