Update colbert.py (#18)

mosheber · web-flow · commit bf4df83b1a58 · 2023-06-13T13:35:17.000+03:00
add checkpoint loading to colbert ranker
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,
