mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
fix: Reorganize the model cache (#3315)
This commit is contained in:
parent
a9e9f5b085
commit
a9c46cd7e0
|
|
@ -22,6 +22,9 @@ class XInferenceReranker(MaxKBBaseModel, BaseDocumentCompressor):
|
||||||
"""UID of the launched model"""
|
"""UID of the launched model"""
|
||||||
api_key: Optional[str]
|
api_key: Optional[str]
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def is_cache_model():
|
||||||
|
return False
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
|
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
|
||||||
return XInferenceReranker(server_url=model_credential.get('server_url'), model_uid=model_name,
|
return XInferenceReranker(server_url=model_credential.get('server_url'), model_uid=model_name,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue