primerz commited on
Commit
b548850
·
verified ·
1 Parent(s): 66b6852

Update cappella.py

Browse files
Files changed (1) hide show
  1. 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.pooler_output
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.