From 9231143f91863ffbe0542bc69a90b723a40e165d Mon Sep 17 00:00:00 2001 From: Zander Chase <130414180+vowelparrot@users.noreply.github.com> Date: Wed, 10 May 2023 10:21:54 -0700 Subject: [PATCH] Fix Duplicate trust_remote_code in pipeline (#4369) ### Fix issue with duplicate specification of `trust_remote_code` in HuggingFacePipeline Fixes # 4351 --- langchain/llms/huggingface_pipeline.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/langchain/llms/huggingface_pipeline.py b/langchain/llms/huggingface_pipeline.py index 529cea28cc..9d63aa0432 100644 --- a/langchain/llms/huggingface_pipeline.py +++ b/langchain/llms/huggingface_pipeline.py @@ -115,7 +115,10 @@ class HuggingFacePipeline(LLM): "can be a positive integer associated with CUDA device id.", cuda_device_count, ) - + if "trust_remote_code" in _model_kwargs: + _model_kwargs = { + k: v for k, v in _model_kwargs.items() if k != "trust_remote_code" + } pipeline = hf_pipeline( task=task, model=model,