diff --git a/evofactory/models/causallm.py b/evofactory/models/causallm.py index 58ecde3..37c9f99 100644 --- a/evofactory/models/causallm.py +++ b/evofactory/models/causallm.py @@ -82,7 +82,7 @@ def __init__(self, model_path: str, **kwargs): super().__init__(**kwargs) from vllm import LLM - self.model = LLM(model=model_path) + self.model = LLM(model=model_path, **self.model_kwargs) self.post_init() def post_init(self):