diff --git a/libs/community/langchain_community/llms/huggingface_pipeline.py b/libs/community/langchain_community/llms/huggingface_pipeline.py index a1989702b0e..1ccbc76954a 100644 --- a/libs/community/langchain_community/llms/huggingface_pipeline.py +++ b/libs/community/langchain_community/llms/huggingface_pipeline.py @@ -261,7 +261,10 @@ class HuggingFacePipeline(BaseLLM): ) -> LLMResult: # List to hold all results text_generations: List[str] = [] - pipeline_kwargs = kwargs.get("pipeline_kwargs", {}) + + default_pipeline_kwargs = self.pipeline_kwargs if self.pipeline_kwargs else {} + pipeline_kwargs = kwargs.get("pipeline_kwargs", default_pipeline_kwargs) + skip_prompt = kwargs.get("skip_prompt", False) for i in range(0, len(prompts), self.batch_size):