From db3583e7f29693ec5eecd9895d03bace4367bc1a Mon Sep 17 00:00:00 2001 From: Richie Cahill Date: Tue, 14 Apr 2026 20:22:02 -0400 Subject: [PATCH] removed richie dot files refrens --- pipelines/tools/containers/finetune.py | 21 +++++++++++++-------- 1 file changed, 13 insertions(+), 8 deletions(-) diff --git a/pipelines/tools/containers/finetune.py b/pipelines/tools/containers/finetune.py index bece28a..f28fe02 100644 --- a/pipelines/tools/containers/finetune.py +++ b/pipelines/tools/containers/finetune.py @@ -15,7 +15,7 @@ logger = logging.getLogger(__name__) CONTAINER_NAME = "bill-finetune" FINETUNE_IMAGE = "bill-finetune:latest" -DOCKERFILE_PATH = "/home/richie/dotfiles/python/prompt_bench/Dockerfile.finetune" +REPO_DIR = Path(__file__).resolve().parents[4] DEFAULT_HF_CACHE = Path("/zfs/models/hf") @@ -23,7 +23,15 @@ def build_image() -> None: """Build the fine-tuning Docker image.""" logger.info("Building fine-tuning image: %s", FINETUNE_IMAGE) result = subprocess.run( - ["docker", "build", "-f", DOCKERFILE_PATH, "-t", FINETUNE_IMAGE, "."], + [ + "docker", + "build", + "-f", + str(REPO_DIR / "python/prompt_bench/Dockerfile.finetune"), + "-t", + FINETUNE_IMAGE, + ".", + ], text=True, check=False, ) @@ -124,14 +132,11 @@ def build() -> None: @app.command() def run( - dataset: Annotated[Path, typer.Option(help="Fine-tuning JSONL")] = Path( - "/home/richie/dotfiles/data/finetune_dataset.jsonl" - ), + dataset: Annotated[Path, typer.Option(help="Fine-tuning JSONL")] = REPO_DIR + / "data/finetune_dataset.jsonl", output_dir: Annotated[ Path, typer.Option(help="Where to save the trained model") - ] = Path( - "/home/richie/dotfiles/data/output/qwen-bill-summarizer", - ), + ] = REPO_DIR / "data/output/qwen-bill-summarizer", hf_cache: Annotated[ Path, typer.Option(help="Host path to HuggingFace model cache") ] = DEFAULT_HF_CACHE,