修复 model_provider_lite 的 bug
This commit is contained in:
parent
a20143b09c
commit
42fa8c19a1
@ -162,8 +162,8 @@ class Retriever:
|
|||||||
def reco_entities(self, query, history, refs):
|
def reco_entities(self, query, history, refs):
|
||||||
"""识别句子中的实体"""
|
"""识别句子中的实体"""
|
||||||
query = refs.get("rewritten_query", query)
|
query = refs.get("rewritten_query", query)
|
||||||
model_provider = config.model_provider_lite
|
model_provider = config.model_provider
|
||||||
model_name = config.model_name_lite
|
model_name = config.model_name
|
||||||
model = select_model(model_provider=model_provider, model_name=model_name)
|
model = select_model(model_provider=model_provider, model_name=model_name)
|
||||||
|
|
||||||
entities = []
|
entities = []
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user