From 12b1376ba12b69398b47f6a18ec83e0604b68e34 Mon Sep 17 00:00:00 2001 From: Chris Hua Date: Tue, 30 May 2023 19:12:34 -0400 Subject: [PATCH] fix: various PR's https://github.com/stanford-futuredata/ColBERT/pull/180 https://github.com/stanford-futuredata/ColBERT/pull/112 --- src/colbert/index_updater.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/colbert/index_updater.py b/src/colbert/index_updater.py index 84ba2ff2..4e2c40ba 100644 --- a/src/colbert/index_updater.py +++ b/src/colbert/index_updater.py @@ -390,7 +390,7 @@ def _add_pid_to_ivf(self, partitions, pid): assert sum(new_ivf_lengths) == len(new_ivf) # Replace the current ivf with new_ivf - self.curr_ivf = torch.tensor(new_ivf) + self.curr_ivf = torch.tensor(new_ivf, dtype=torch.int32) self.curr_ivf_lengths = torch.tensor(new_ivf_lengths) def _write_to_last_chunk(self, pid_start, pid_end, emb_start, emb_end):