Tree: Format
Signed-off-by: kingbri <8082010+kingbri1@users.noreply.github.com>
This commit is contained in:
parent
3f09fcd8c9
commit
f2c7da2faf
2 changed files with 1 additions and 4 deletions
|
|
@ -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",
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue