FROM ghcr.io/unslothai/unsloth:latest

RUN pip install --no-cache-dir typer

WORKDIR /workspace
COPY python/prompt_bench/finetune.py python/prompt_bench/finetune.py
COPY config/prompts/summarization_prompts.toml config/prompts/summarization_prompts.toml
COPY python/prompt_bench/__init__.py python/prompt_bench/__init__.py
COPY python/__init__.py python/__init__.py

ENTRYPOINT ["python", "-m", "pipelines.prompt_bench.finetune"]