github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-02 11:39:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

huggingface: fix community dep checking (#21628)

Browse Source
This commit is contained in:
Erick Friis
2024-05-13 14:52:18 -07:00
committed by GitHub
parent 91a2ea5cd6
commit 9b51ca08bc
6 changed files with 42 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/partners/huggingface/langchain_huggingface/llms/huggingface_pipeline.py
View File

@@ -33,7 +33,7 @@ class HuggingFacePipeline(BaseLLM):
Example using from_model_id:
.. code-block:: python
from langchain_community.llms import HuggingFacePipeline
from langchain_huggingface import HuggingFacePipeline
hf = HuggingFacePipeline.from_model_id(
model_id="gpt2",
task="text-generation",
@@ -42,7 +42,7 @@ class HuggingFacePipeline(BaseLLM):
Example passing pipeline in directly:
.. code-block:: python
from langchain_community.llms import HuggingFacePipeline
from langchain_huggingface import HuggingFacePipeline
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
model_id = "gpt2"