from app.engines.base import BaseEngine


class VisionEngine(BaseEngine):
    """Training engine for ViT, CLIP, and other vision models."""

    async def load_model(self, model_id: str, **kwargs):
        raise NotImplementedError

    def get_peft_config(self, method: str, params: dict):
        raise NotImplementedError

    async def preprocess_dataset(self, dataset_path: str, output_path: str, **kwargs):
        raise NotImplementedError

    async def train(self, job_id: str, dataset_path: str, peft_config, training_args: dict):
        raise NotImplementedError

    def get_model_info(self, model_id: str):
        raise NotImplementedError