From 82974d9550eaefdcadc2a6ec9a9e346b7c876942 Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Sun, 6 Apr 2025 20:47:45 +0800 Subject: [PATCH] fix typo found by copilot --- src/models/embedding.py | 2 +- src/models/rerank_model.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/models/embedding.py b/src/models/embedding.py index ff1a3bcf..aa035bc8 100644 --- a/src/models/embedding.py +++ b/src/models/embedding.py @@ -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) diff --git a/src/models/rerank_model.py b/src/models/rerank_model.py index a2fb7d37..145ad2b4 100644 --- a/src/models/rerank_model.py +++ b/src/models/rerank_model.py @@ -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()}")