diff --git a/libs/community/langchain_community/llms/bigdl_llm.py b/libs/community/langchain_community/llms/bigdl_llm.py index bcc16dbb7e85d..3181e5f540eeb 100644 --- a/libs/community/langchain_community/llms/bigdl_llm.py +++ b/libs/community/langchain_community/llms/bigdl_llm.py @@ -22,10 +22,11 @@ class BigdlLLM(IpexLLM): def from_model_id( cls, model_id: str, + model_kwargs: Optional[dict] = None, + *, tokenizer_id: Optional[str] = None, load_in_4bit: bool = True, load_in_low_bit: Optional[str] = None, - model_kwargs: Optional[dict] = None, **kwargs: Any, ) -> LLM: """ @@ -105,8 +106,9 @@ def from_model_id( def from_model_id_low_bit( cls, model_id: str, - tokenizer_id: Optional[str] = None, model_kwargs: Optional[dict] = None, + *, + tokenizer_id: Optional[str] = None, **kwargs: Any, ) -> LLM: """ diff --git a/libs/community/langchain_community/llms/ipex_llm.py b/libs/community/langchain_community/llms/ipex_llm.py index ceb6f79cca8c2..a8e60fe6b1f4a 100644 --- a/libs/community/langchain_community/llms/ipex_llm.py +++ b/libs/community/langchain_community/llms/ipex_llm.py @@ -41,10 +41,11 @@ class Config: def from_model_id( cls, model_id: str, + model_kwargs: Optional[dict] = None, + *, tokenizer_id: Optional[str] = None, load_in_4bit: bool = True, load_in_low_bit: Optional[str] = None, - model_kwargs: Optional[dict] = None, **kwargs: Any, ) -> LLM: """ @@ -82,8 +83,9 @@ def from_model_id( def from_model_id_low_bit( cls, model_id: str, - tokenizer_id: Optional[str] = None, model_kwargs: Optional[dict] = None, + *, + tokenizer_id: Optional[str] = None, **kwargs: Any, ) -> LLM: """