dzenzzz commited on
Commit
84fd3b1
·
1 Parent(s): 2acaa2a

minor change

Browse files
Files changed (1) hide show
  1. neural_searcher.py +1 -1
neural_searcher.py CHANGED
@@ -17,7 +17,7 @@ class NeuralSearcher:
17
 
18
  dense_query = next(iter(self.dense_model.encode(text)))
19
  sparse_query = next(self.sparse_model.query_embed(text))
20
- late_query = next(self.late_interaction_embedding_model.query_embed(text))
21
 
22
  prefetch = [
23
  models.Prefetch(
 
17
 
18
  dense_query = next(iter(self.dense_model.encode(text)))
19
  sparse_query = next(self.sparse_model.query_embed(text))
20
+ late_query = next(self.late_interaction_model.query_embed(text))
21
 
22
  prefetch = [
23
  models.Prefetch(