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,