No mask in indexer forward (#2)

osalpekar · Feb 7, 2024 · befeda4 · befeda4
1 parent 0485c9f
commit befeda4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/indexer.py b/indexer.py
@@ -96,7 +96,7 @@ def index(self):
                 #     inputs, output_hidden_states=True
                 # )
                 _, embedding = self.model.forward(
-                    tokens, 0, output_last_hidden_state=True, mask=attn_mask
+                    tokens, 0, output_last_hidden_state=True
                 )
                 del attn_mask
                 del tokens