12 lines
430 B
Docker
12 lines
430 B
Docker
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"]
|