jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

remove unnecessary code

Browse source
This commit is contained in:
TerminalMan 2024-09-14 22:49:37 +01:00
parent dc4946b565
commit 533e7c9119
1 changed files with 2 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

14
common/config_models.py
View file

@ -1,8 +1,6 @@
from pydantic import BaseModel, ConfigDict, Field, model_validator
from pydantic import BaseModel, ConfigDict, Field
from typing import List, Literal, Optional, Union
from common.utils import unwrap
CACHE_SIZES = Literal["FP16", "Q8", "Q6", "Q4"]
@ -311,14 +309,6 @@ class TabbyConfigModel(BaseModel):
default_factory=EmbeddingsConfig.model_construct
)
@model_validator(mode="before")
def set_defaults(cls, values):
for field_name, field_value in values.items():
if field_value is None:
default_instance = cls.__annotations__[field_name]().dict()
values[field_name] = cls.__annotations__[field_name](**default_instance)
return values
model_config = ConfigDict(validate_assignment=True, protected_namespaces=())
@ -335,7 +325,7 @@ def generate_config_file(filename="config_sample.yml", indentation=2):
comment = section[property]["description"]
yaml += f"{indent}# {comment}\n"
value = unwrap(section[property].get("default"), "")
value = section[property].get("default", "")
yaml += f"{indent}{property}: {value}\n\n"
return yaml + "\n"