|
@@ -206,7 +206,7 @@ def update_reranking_model(
|
|
|
self.device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
self.ckpt = Checkpoint(
|
|
|
get_model_path(name, auto_update),
|
|
|
- colbert_config=ColBERTConfig(),
|
|
|
+ colbert_config=ColBERTConfig(model_name=name),
|
|
|
).to(self.device)
|
|
|
pass
|
|
|
|