Skip to content

Commit

Permalink
fixed property of index_config
Browse files Browse the repository at this point in the history
  • Loading branch information
mlomeli1 committed Nov 28, 2023
1 parent 23639ea commit 0ec8889
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/index.py
Original file line number Diff line number Diff line change
Expand Up @@ -287,8 +287,7 @@ def _set_index_config_options(self, index_config: GPUIndexConfig) -> GPUIndexCon
"""
index_config.device = torch.cuda.current_device()
index_config.indicesOptions = faiss.INDICES_32_BIT
index_config.useFloat16 = True

index_config.useFloat16LookupTables = True
return index_config

def _create_PQ_index(self, dimension) -> FAISSIndex:
Expand Down

0 comments on commit 0ec8889

Please sign in to comment.