Timothy J. Baek hai 7 meses
pai
achega
b7f0759485
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      backend/open_webui/apps/rag/main.py

+ 1 - 1
backend/open_webui/apps/rag/main.py

@@ -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: