kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Wenjie Zhang 2024-11-22 00:04:30 +08:00
parent 0faa2be658
commit c51c9b152d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/models/embedding.py
View File

@ -20,13 +20,13 @@ class EmbeddingModel(FlagModel):
model_name=model_info["name"], model_name=model_info["name"],
default_path=model_info.get("default_path", None)) default_path=model_info.get("default_path", None))
logger.info(f"Loading embedding model {model_info["name"]} from {model_name_or_path}") logger.info(f"Loading embedding model {model_info['name']} from {model_name_or_path}")
super().__init__(model_name_or_path, super().__init__(model_name_or_path,
query_instruction_for_retrieval=model_info.get("query_instruction", None), query_instruction_for_retrieval=model_info.get("query_instruction", None),
use_fp16=False, **kwargs) use_fp16=False, **kwargs)
logger.info(f"Embedding model {model_info["name"]} loaded") logger.info(f"Embedding model {model_info['name']} loaded")
class Reranker(FlagReranker): class Reranker(FlagReranker):