removed richie dot files refrens
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user