Tree: Format

Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
This commit is contained in:
kingbri 2025-04-21 23:16:14 -04:00
parent 3f09fcd8c9
commit f2c7da2faf
2 changed files with 1 additions and 4 deletions

View file

@ -188,7 +188,6 @@ class BaseModelContainer(abc.ABC):
if False: if False:
yield yield
# TODO: Replace by yielding a model card
@abc.abstractmethod @abc.abstractmethod
def model_info(self) -> ModelCard: def model_info(self) -> ModelCard:
""" """

View file

@ -241,9 +241,7 @@ async def find_prompt_template(template_name, model_dir: pathlib.Path):
# Add lookup from prompt template name if provided # Add lookup from prompt template name if provided
if template_name: if template_name:
find_template_functions[:0] = [ find_template_functions[:0] = [
lambda: PromptTemplate.from_file( lambda: PromptTemplate.from_file(pathlib.Path("templates") / template_name),
pathlib.Path("templates") / template_name
),
lambda: PromptTemplate.from_model_json( lambda: PromptTemplate.from_model_json(
model_dir / "tokenizer_config.json", model_dir / "tokenizer_config.json",
key="chat_template", key="chat_template",