Templating: Support chat_template.jinja
This commit is contained in:
parent
03d72a37be
commit
81a115b781
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.")
|
logger.info("Attempting to load a prompt template if present.")
|
||||||
|
|
||||||
find_template_functions = [
|
find_template_functions = [
|
||||||
|
lambda: PromptTemplate.from_file(model_dir / "chat_template.jinja"),
|
||||||
lambda: PromptTemplate.from_model_json(
|
lambda: PromptTemplate.from_model_json(
|
||||||
model_dir / "chat_template.json",
|
model_dir / "chat_template.json",
|
||||||
key="chat_template",
|
key="chat_template",
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue