From f2c7da2faf419a8fffca7ca4c22bccfe616b0ba0 Mon Sep 17 00:00:00 2001 From: kingbri <8082010+kingbri1@users.noreply.github.com> Date: Mon, 21 Apr 2025 23:16:14 -0400 Subject: [PATCH] Tree: Format Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com> --- backends/base_model_container.py | 1 - common/templating.py | 4 +--- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/backends/base_model_container.py b/backends/base_model_container.py index 1eeae5a..aca3a77 100644 --- a/backends/base_model_container.py +++ b/backends/base_model_container.py @@ -188,7 +188,6 @@ class BaseModelContainer(abc.ABC): if False: yield - # TODO: Replace by yielding a model card @abc.abstractmethod def model_info(self) -> ModelCard: """ diff --git a/common/templating.py b/common/templating.py index ff21300..d71de11 100644 --- a/common/templating.py +++ b/common/templating.py @@ -241,9 +241,7 @@ async def find_prompt_template(template_name, model_dir: pathlib.Path): # Add lookup from prompt template name if provided if template_name: find_template_functions[:0] = [ - lambda: PromptTemplate.from_file( - pathlib.Path("templates") / template_name - ), + lambda: PromptTemplate.from_file(pathlib.Path("templates") / template_name), lambda: PromptTemplate.from_model_json( model_dir / "tokenizer_config.json", key="chat_template",