This commit is contained in:
Timothy J. Baek 2024-09-16 12:36:43 +02:00
parent bc6f23f82f
commit b7f0759485

View File

@ -199,7 +199,7 @@ def update_reranking_model(
update_model: bool = False,
):
if reranking_model:
if reranking_model in ["jinaai/jina-colbert-v2"]:
if any(model in reranking_model for model in ["jinaai/jina-colbert-v2"]):
class Colbert:
def __init__(self, name) -> None: