From b53559de6f25ff32180e60e09313275ce216d04d Mon Sep 17 00:00:00 2001 From: Aaron <29749331+aarnphm@users.noreply.github.com> Date: Thu, 30 Nov 2023 07:36:34 -0500 Subject: [PATCH] fix(setter): correct item with the same kwargs with stubs Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com> --- openllm-python/src/openllm/_llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openllm-python/src/openllm/_llm.py b/openllm-python/src/openllm/_llm.py index 8bc35318..f9348314 100644 --- a/openllm-python/src/openllm/_llm.py +++ b/openllm-python/src/openllm/_llm.py @@ -257,7 +257,7 @@ class LLM(t.Generic[M, T]): @property def _model_attrs(self): return {**self.import_kwargs[0], **self.__model_attrs} @_model_attrs.setter - def _model_attrs(self, value): self.__model_attrs = value + def _model_attrs(self, model_attrs): self.__model_attrs = model_attrs @property def _tokenizer_attrs(self): return {**self.import_kwargs[1], **self.__tokenizer_attrs} def _cascade_backend(self) -> LiteralBackend: