27 lines
991 B
Docker
27 lines
991 B
Docker
# Unsloth fine-tuning container for Qwen 3.5 4B on RTX 3090.
|
|
#
|
|
# Build:
|
|
# docker build -f pipelines/pipelines/tools/Dockerfile.finetune -t bill-finetune .
|
|
#
|
|
# Run:
|
|
# docker run --rm --device=nvidia.com/gpu=all --ipc=host \
|
|
# -v $(pwd)/output:/workspace/output \
|
|
# -v $(pwd)/output/finetune_dataset.jsonl:/workspace/dataset.jsonl:ro \
|
|
# -v /zfs/models/hf:/models \
|
|
# bill-finetune \
|
|
# --dataset /workspace/dataset.jsonl \
|
|
# --output-dir /workspace/output/qwen-bill-summarizer
|
|
|
|
FROM ghcr.io/unslothai/unsloth:latest
|
|
|
|
RUN pip install --no-cache-dir typer rouge-score
|
|
|
|
WORKDIR /workspace
|
|
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"]
|