Nornsight/weave
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add eval rouge metrics

Browse Source
This commit is contained in:
Matthew Michal
2026-04-19 23:40:50 -04:00
parent db3583e7f2
commit 1e9c2a6caa
4 changed files with 435 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pipelines/tools/Dockerfile.finetune
+2 -1
View File
@@ -14,12 +14,13 @@
FROM ghcr.io/unslothai/unsloth:latest
RUN pip install --no-cache-dir typer
RUN pip install --no-cache-dir typer rouge-score
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/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"]