phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

upgrade pydantic to latest

Browse source
This commit is contained in:
Hardik Shah 2024-08-12 15:14:21 -07:00
parent 2cd8b2ff5b
commit 37da47ef8e
4 changed files with 22 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_toolchain/inference/meta_reference/config.py
View file

@ -11,9 +11,9 @@ from llama_models.datatypes import ModelFamily
from llama_models.schema_utils import json_schema_type
from llama_models.sku_list import all_registered_models
from llama_toolchain.inference.api import QuantizationConfig
from pydantic import BaseModel, Field, field_validator
from pydantic import BaseModel, Field, validator
from llama_toolchain.inference.api import QuantizationConfig
@json_schema_type
@ -27,7 +27,7 @@ class MetaReferenceImplConfig(BaseModel):
max_seq_len: int
max_batch_size: int = 1
@validator("model")
@field_validator("model")
@classmethod
def validate_model(cls, model: str) -> str:
permitted_models = [