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

fix typo found by copilot

Browse Source
This commit is contained in:
Wenjie Zhang 2025-04-06 20:47:45 +08:00
parent 1d40da7460
commit 82974d9550
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/models/embedding.py
View File

@ -15,7 +15,7 @@ class BaseEmbeddingModel:
if hasattr(self, "dimension"):
return self.dimension
if hasattr("embed_model_fullname"):
if hasattr(self, "embed_model_fullname"):
return config.embed_model_names[self.embed_model_fullname].get("dimension", None)
return config.embed_model_names[self.model].get("dimension", None)

4
src/models/rerank_model.py
View File

@ -22,7 +22,7 @@ class LocalReranker(FlagReranker):
def sigmoid(x):
return 1 / (1 + np.exp(-x))
class SilconFlowReranker():
class SiliconFlowReranker():
def __init__(self, config, **kwargs):
self.url = "https://api.siliconflow.cn/v1/rerank"
self.model = config.reranker_names[config.reranker]["name"]
@ -64,7 +64,7 @@ def get_reranker(config):
if provider == "local":
return LocalReranker(config)
elif provider == "siliconflow":
return SilconFlowReranker(config)
return SiliconFlowReranker(config)
else:
raise ValueError(f"Unsupported Reranker: {config.reranker}, only support {config.reranker_names.keys()}")