fix docstring (#383)

![Screenshot 2022-12-19 at 11 06 48
AM](https://user-images.githubusercontent.com/43734688/208468970-5cb9bafb-f535-486e-b41f-312a2f9ffffb.png)
This commit is contained in:
mrbean 2022-12-19 11:10:17 -05:00 committed by GitHub
parent a01d3e6955
commit fc66a32c6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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