diff --git a/langchain/llms/replicate.py b/langchain/llms/replicate.py index f4660f6b37..5e932ca565 100644 --- a/langchain/llms/replicate.py +++ b/langchain/llms/replicate.py @@ -72,6 +72,7 @@ class Replicate(LLM): def _identifying_params(self) -> Mapping[str, Any]: """Get the identifying parameters.""" return { + "model": self.model, **{"model_kwargs": self.model_kwargs}, }