from pydantic import BaseModel


class ModelConfig(BaseModel):
    model_name: str
    model_path: str
    format: str = "tensorrt"
    precision: str = "fp16"
    warmup_iterations: int = 3