slight upgrade to CLI

This commit is contained in:
Ashwin Bharambe 2024-10-06 18:02:47 -07:00 committed by Ashwin Bharambe
parent 1550187cd8
commit 099a95b614
4 changed files with 23 additions and 21 deletions

View file

@ -34,7 +34,7 @@ class InferenceEndpointImplConfig(BaseModel):
@json_schema_type
class InferenceAPIImplConfig(BaseModel):
model_id: str = Field(
huggingface_repo: str = Field(
description="The model ID of the model on the Hugging Face Hub (e.g. 'meta-llama/Meta-Llama-3.1-70B-Instruct')",
)
api_token: Optional[str] = Field(