huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 04:14:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix warning for pydantic 2

Browse Source
This commit is contained in:
Tim Shih 2023-12-01 10:37:14 +08:00
parent ccd5725a0c
commit 75e8521eb9
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clients/python/text_generation/types.py
View File

@ -243,5 +243,11 @@ class StreamResponse(BaseModel):
# Inference API currently deployed model # Inference API currently deployed model
class DeployedModel(BaseModel): class DeployedModel(BaseModel):
model_config = {
# Pydantic 2 sets `model_` prefix as protected namespace and raises a
# warning when this model is loaded. Set this option to disable the warning.
"protected_namespaces": (),
}
model_id: str model_id: str
sha: str sha: str