Update colbert.py (#18)

add checkpoint loading to colbert ranker
IntelLabs · Jun 13, 2023 · bf4df83 · bf4df83
1 parent 0282626
commit bf4df83
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/fastrag/rankers/colbert.py b/fastrag/rankers/colbert.py
@@ -29,6 +29,7 @@ def __init__(
             self.devices, _ = initialize_device_settings(use_cuda=use_gpu, multi_gpu=True)
 
         self.config = ColBERTConfig(
+            checkpoint=checkpoint_path,
             ncells=2,
             nbits=2,
             query_maxlen=query_maxlen,