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