diff --git a/pipelines/tools/Dockerfile.finetune b/pipelines/tools/Dockerfile.finetune index c33d4a1..862f325 100644 --- a/pipelines/tools/Dockerfile.finetune +++ b/pipelines/tools/Dockerfile.finetune @@ -17,9 +17,10 @@ FROM ghcr.io/unslothai/unsloth:latest RUN pip install --no-cache-dir typer rouge-score WORKDIR /workspace -COPY pipelines/pipelines/tools/__init__.py pipelines/tools/__init__.py -COPY pipelines/pipelines/tools/finetune.py pipelines/tools/finetune.py -COPY pipelines/pipelines/tools/summarization_eval.py pipelines/tools/summarization_eval.py -COPY config/prompts/summarization_prompts.toml config/prompts/summarization_prompts.toml +COPY pipelines/tools/__init__.py pipelines/tools/__init__.py +COPY pipelines/tools/finetune.py pipelines/tools/finetune.py +COPY pipelines/tools/summarization_eval.py pipelines/tools/summarization_eval.py +COPY summarization_prompts.toml config/prompts/summarization_prompts.toml +COPY config.toml pipelines/tools/config.toml ENTRYPOINT ["python", "-m", "pipelines.tools.finetune"]