mirror of
https://github.com/theroyallab/tabbyAPI.git
synced 2026-03-15 00:07:28 +00:00
Tree: Format
Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
This commit is contained in:
@@ -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:
|
||||||
"""
|
"""
|
||||||
|
|||||||
@@ -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",
|
||||||
|
|||||||
Reference in New Issue
Block a user