testing/pipelines
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed path

Browse Source
This commit is contained in:
Richie Cahill
2026-04-19 23:50:10 -04:00
parent 1e9c2a6caa
commit 716bed5300
2 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pipelines/tools/Dockerfile.finetune
View File

@@ -1,7 +1,7 @@
# Unsloth fine-tuning container for Qwen 3.5 4B on RTX 3090. # Unsloth fine-tuning container for Qwen 3.5 4B on RTX 3090.
# #
# Build: # Build:
# docker build -f python/prompt_bench/Dockerfile.finetune -t bill-finetune . # docker build -f pipelines/pipelines/tools/Dockerfile.finetune -t bill-finetune .
# #
# Run: # Run:
# docker run --rm --device=nvidia.com/gpu=all --ipc=host \ # docker run --rm --device=nvidia.com/gpu=all --ipc=host \
@@ -17,10 +17,9 @@ FROM ghcr.io/unslothai/unsloth:latest
RUN pip install --no-cache-dir typer rouge-score RUN pip install --no-cache-dir typer rouge-score
WORKDIR /workspace WORKDIR /workspace
COPY python/prompt_bench/finetune.py python/prompt_bench/finetune.py 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 config/prompts/summarization_prompts.toml config/prompts/summarization_prompts.toml
COPY python/prompt_bench/__init__.py python/prompt_bench/__init__.py
COPY python/prompt_bench/summarization_eval.py python/prompt_bench/summarization_eval.py
COPY python/__init__.py python/__init__.py
ENTRYPOINT ["python", "-m", "pipelines.prompt_bench.finetune"] ENTRYPOINT ["python", "-m", "pipelines.tools.finetune"]

2
pipelines/tools/containers/finetune.py
View File

@@ -27,7 +27,7 @@ def build_image() -> None:
"docker", "docker",
"build", "build",
"-f", "-f",
str(REPO_DIR / "python/prompt_bench/Dockerfile.finetune"), str(REPO_DIR / "pipelines/pipelines/tools/Dockerfile.finetune"),
"-t", "-t",
FINETUNE_IMAGE, FINETUNE_IMAGE,
".", ".",