Merge pull request #374 from DocShotgun/main
Templating: Support chat_template.jinja
This commit is contained in:
commit
83f778db2d
1 changed files with 1 additions and 0 deletions
|
|
@ -218,6 +218,7 @@ async def find_prompt_template(template_name, model_dir: pathlib.Path):
|
|||
logger.info("Attempting to load a prompt template if present.")
|
||||
|
||||
find_template_functions = [
|
||||
lambda: PromptTemplate.from_file(model_dir / "chat_template.jinja"),
|
||||
lambda: PromptTemplate.from_model_json(
|
||||
model_dir / "chat_template.json",
|
||||
key="chat_template",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue