Update the default value for TGI URL

This commit is contained in:
Celina Hanouti 2024-09-05 19:05:07 +02:00
parent e5bcfdac21
commit 7aa50934bf

View file

@ -13,7 +13,7 @@ from pydantic import BaseModel, Field, field_validator
@json_schema_type @json_schema_type
class TGIImplConfig(BaseModel): class TGIImplConfig(BaseModel):
url: str = Field( url: str = Field(
default="https://api-inference.huggingface.co", default="https://huggingface.co/inference-endpoints/dedicated",
description="The URL for the TGI endpoint", description="The URL for the TGI endpoint",
) )
api_token: Optional[str] = Field( api_token: Optional[str] = Field(