corrected Pydantic warning.

This commit is contained in:
jeff 2024-06-19 21:33:39 -04:00
parent cdbf802860
commit f1a6fdb900

View File

@ -1,5 +1,5 @@
from enum import Enum
from pydantic import BaseModel, field_validator
from pydantic import BaseModel, field_validator, ConfigDict
from typing import Optional, List, Union, Any
from text_generation.errors import ValidationError
@ -452,5 +452,6 @@ class StreamResponse(BaseModel):
# Inference API currently deployed model
class DeployedModel(BaseModel):
model_config = ConfigDict(protected_namespaces=())
model_id: str
sha: str