Skip to content

Commit 83f778d

Browse files
authored
Merge pull request #374 from DocShotgun/main
Templating: Support chat_template.jinja
2 parents 03d72a3 + 81a115b commit 83f778d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

common/templating.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,7 @@ async def find_prompt_template(template_name, model_dir: pathlib.Path):
218218
logger.info("Attempting to load a prompt template if present.")
219219

220220
find_template_functions = [
221+
lambda: PromptTemplate.from_file(model_dir / "chat_template.jinja"),
221222
lambda: PromptTemplate.from_model_json(
222223
model_dir / "chat_template.json",
223224
key="chat_template",

0 commit comments

Comments
 (0)