diff --git a/langchain/llms/huggingface_pipeline.py b/langchain/llms/huggingface_pipeline.py index 84b44c766b..eab476ce9f 100644 --- a/langchain/llms/huggingface_pipeline.py +++ b/langchain/llms/huggingface_pipeline.py @@ -26,7 +26,7 @@ class HuggingFacePipeline(LLM, BaseModel): model_id="gpt2", task="text-generation" ) Example passing pipeline in directly: - .. code-block:: python + .. code-block:: python from langchain.llms.huggingface_pipeline import HuggingFacePipeline from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline