mirror of
https://github.com/theroyallab/tabbyAPI.git
synced 2026-04-28 02:01:24 +00:00
Merge pull request #374 from DocShotgun/main
Templating: Support chat_template.jinja
This commit is contained in:
@@ -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",
|
||||||
|
|||||||
Reference in New Issue
Block a user