mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-20 13:54:48 +00:00
fix docstring (#383)

This commit is contained in:
parent
a01d3e6955
commit
fc66a32c6f
@ -26,7 +26,7 @@ class HuggingFacePipeline(LLM, BaseModel):
|
|||||||
model_id="gpt2", task="text-generation"
|
model_id="gpt2", task="text-generation"
|
||||||
)
|
)
|
||||||
Example passing pipeline in directly:
|
Example passing pipeline in directly:
|
||||||
.. code-block:: python
|
.. code-block:: python
|
||||||
|
|
||||||
from langchain.llms.huggingface_pipeline import HuggingFacePipeline
|
from langchain.llms.huggingface_pipeline import HuggingFacePipeline
|
||||||
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
||||||
|
Loading…
Reference in New Issue
Block a user