This commit is contained in:
Wauplin 2024-09-25 22:27:06 +02:00
parent 854ff85784
commit fcb3438031
No known key found for this signature in database
GPG key ID: 9838FE02BECE1A02
3 changed files with 15 additions and 5 deletions

View file

@ -20,6 +20,7 @@ class TGIImplConfig(BaseModel):
description="A bearer token if your TGI endpoint is protected.",
)
@json_schema_type
class InferenceEndpointImplConfig(BaseModel):
endpoint_name: str = Field(
@ -40,5 +41,3 @@ class InferenceAPIImplConfig(BaseModel):
default=None,
description="Your Hugging Face user access token (will default to locally saved token if not provided)",
)