Spaces:
Runtime error
Runtime error
Update cappella.py
Browse files- cappella.py +1 -1
cappella.py
CHANGED
|
@@ -86,7 +86,7 @@ class Cappella:
|
|
| 86 |
output_hidden_states=True
|
| 87 |
)
|
| 88 |
embeds_2 = output_2.hidden_states[-2]
|
| 89 |
-
pooled_embeds = output_2.
|
| 90 |
|
| 91 |
# --- Concatenate ---
|
| 92 |
# The final embeddings are a concatenation of both.
|
|
|
|
| 86 |
output_hidden_states=True
|
| 87 |
)
|
| 88 |
embeds_2 = output_2.hidden_states[-2]
|
| 89 |
+
pooled_embeds = output_2.text_embeds
|
| 90 |
|
| 91 |
# --- Concatenate ---
|
| 92 |
# The final embeddings are a concatenation of both.
|