automatic/scripts/lbm/base/model_config.py

from pydantic.dataclasses import dataclass
from ..config import BaseConfig


@dataclass
class ModelConfig(BaseConfig):
    input_key: str = "image"