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",