Fix HuggingFace model error. (#3870)

### What problem does this PR solve?

#3865

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2024-12-05 13:28:42 +08:00
committed by GitHub
parent a88a1848ff
commit 593ffc4067
3 changed files with 5 additions and 4 deletions

View File

@ -194,7 +194,7 @@ class LocalAIRerank(Base):
"Content-Type": "application/json",
"Authorization": f"Bearer {key}"
}
self.model_name = model_name.replace("___LocalAI","")
self.model_name = model_name.split("___")[0]
def similarity(self, query: str, texts: list):
# noway to config Ragflow , use fix setting