diff --git a/python/llm/src/bigdl/llm/langchain/llms/bigdlllm.py b/python/llm/src/bigdl/llm/langchain/llms/bigdlllm.py index dce33518..3a606a06 100644 --- a/python/llm/src/bigdl/llm/langchain/llms/bigdlllm.py +++ b/python/llm/src/bigdl/llm/langchain/llms/bigdlllm.py @@ -61,7 +61,7 @@ class BigdlNativeLLM(LLM): Example: .. code-block:: python - from langchain.llms import BigdlNativeLLM + from bigdl.llm.langchain.llms import BigdlNativeLLM llm = BigdlNativeLLM(model_path="/path/to/llama/model") """ @@ -286,7 +286,7 @@ class BigdlNativeLLM(LLM): Example: .. code-block:: python - from langchain.llms import BigdlNativeLLM + from bigdl.llm.langchain.llms import BigdlNativeLLM llm = BigdlNativeLLM(model_path="/path/to/local/llama/model.bin") llm("This is a prompt.") """ @@ -331,7 +331,7 @@ class BigdlNativeLLM(LLM): Example: .. code-block:: python - from langchain.llms import BigdlNativeLLM + from bigdl.llm.langchain.llms import BigdlNativeLLM llm = BigdlNativeLLM( model_path="/path/to/local/model.bin", temperature = 0.5 @@ -364,7 +364,7 @@ class _BaseCausalLM(LLM): Example: .. code-block:: python - from langchain.llms import LlamaLLM + from bigdl.llm.langchain.llms import LlamaLLM llm = LlamaLLM(model_path="/path/to/llama/model") """ @@ -588,7 +588,7 @@ class _BaseCausalLM(LLM): Example: .. code-block:: python - from langchain.llms import LlamaLLM + from bigdl.llm.langchain.llms import LlamaLLM llm = LlamaLLM(model_path="/path/to/local/llama/model.bin") llm("This is a prompt.") """ @@ -636,7 +636,7 @@ class _BaseCausalLM(LLM): Example: .. code-block:: python - from langchain.llms import LlamaLLM + from bigdl.llm.langchain.llms import LlamaLLM llm = LlamaLLM( model_path="/path/to/local/model.bin", temperature = 0.5 diff --git a/python/llm/src/bigdl/llm/langchain/llms/transformersllm.py b/python/llm/src/bigdl/llm/langchain/llms/transformersllm.py index 2fb683a1..85eebf78 100644 --- a/python/llm/src/bigdl/llm/langchain/llms/transformersllm.py +++ b/python/llm/src/bigdl/llm/langchain/llms/transformersllm.py @@ -64,7 +64,7 @@ class TransformersLLM(LLM): Example: .. code-block:: python - from langchain.llms import TransformersLLM + from bigdl.llm.langchain.llms import TransformersLLM llm = TransformersLLM.from_model_id(model_id="THUDM/chatglm-6b") """ diff --git a/python/llm/src/bigdl/llm/langchain/llms/transformerspipelinellm.py b/python/llm/src/bigdl/llm/langchain/llms/transformerspipelinellm.py index a120acf5..0049e841 100644 --- a/python/llm/src/bigdl/llm/langchain/llms/transformerspipelinellm.py +++ b/python/llm/src/bigdl/llm/langchain/llms/transformerspipelinellm.py @@ -66,7 +66,7 @@ class TransformersPipelineLLM(LLM): Example: .. code-block:: python - from langchain.llms import TransformersPipelineLLM + from bigdl.llm.langchain.llms import TransformersPipelineLLM llm = TransformersPipelineLLM.from_model_id(model_id="decapoda-research/llama-7b-hf") """