From 75ed94082524c3dd6c222563e16b8aac8550d508 Mon Sep 17 00:00:00 2001 From: Shengsheng Huang Date: Wed, 27 Mar 2024 01:18:15 +0800 Subject: [PATCH] add bigdl property --- libs/community/langchain_community/llms/bigdl_llm.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/libs/community/langchain_community/llms/bigdl_llm.py b/libs/community/langchain_community/llms/bigdl_llm.py index 28fd8fcb2b72a..a529b2e8c40a2 100644 --- a/libs/community/langchain_community/llms/bigdl_llm.py +++ b/libs/community/langchain_community/llms/bigdl_llm.py @@ -134,4 +134,8 @@ def from_model_id_low_bit( tokenizer=tokenizer, model_kwargs=_model_kwargs, **kwargs, - ) \ No newline at end of file + ) + + @property + def _llm_type(self) -> str: + return "bigdl-llm" \ No newline at end of file