From 6bd6662f5fca5593d1cfdd725707751c2ff075a4 Mon Sep 17 00:00:00 2001 From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com> Date: Tue, 30 Jun 2026 20:41:53 -0400 Subject: [PATCH] docs: rename config and docs to reflect Ornith-35B model --- train-on-this-server.sh | 8 ++++---- training/README.md | 6 +++--- .../configs/{llama2-7b-lora.yaml => ornith-35b-lora.yaml} | 0 3 files changed, 7 insertions(+), 7 deletions(-) rename training/configs/{llama2-7b-lora.yaml => ornith-35b-lora.yaml} (100%) diff --git a/train-on-this-server.sh b/train-on-this-server.sh index 4a6b199..0a19537 100755 --- a/train-on-this-server.sh +++ b/train-on-this-server.sh @@ -70,8 +70,8 @@ fi echo "" echo "=== Starting Training ===" -echo "Config: training/configs/llama2-7b-lora.yaml" -echo "Output: training/output/llama2-7b-lora/" +echo "Config: training/configs/ornith-35b-lora.yaml" +echo "Output: training/output/ornith-35b-lora/" echo "" echo "Training will take 6-24 hours depending on GPU." echo "Press Ctrl+C to stop (model will be saved at checkpoint)." @@ -79,8 +79,8 @@ echo "" # Use torchrun for distributed training (2 GPUs) # Add --check-only to validate without training: -# torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/llama2-7b-lora.yaml --check-only -torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/llama2-7b-lora.yaml +# torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/ornith-35b-lora.yaml --check-only +torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/ornith-35b-lora.yaml echo "" echo "=== Training Complete ===" diff --git a/training/README.md b/training/README.md index c21fee0..d0718ca 100644 --- a/training/README.md +++ b/training/README.md @@ -53,13 +53,13 @@ This splits the 20k dataset into train/test sets (95/5). ### 2. Train Model ```bash -python scripts/train.py --config configs/llama2-7b-lora.yaml +python scripts/train.py --config configs/ornith-35b-lora.yaml ``` Or with custom parameters: ```bash -python scripts/train.py --config configs/llama2-7b-lora.yaml --epochs 5 --batch-size 8 +python scripts/train.py --config configs/ornith-35b-lora.yaml --epochs 5 --batch-size 8 ``` Training takes approximately 6-24 hours depending on GPU. @@ -68,7 +68,7 @@ Training takes approximately 6-24 hours depending on GPU. ```bash python scripts/inference.py \ - --model output/llama2-7b-lora \ + --model output/ornith-35b-lora \ --task "Fix parser crash on malformed JSON" \ --files src/parser.cpp \ --tests-run --test-count 294 \ diff --git a/training/configs/llama2-7b-lora.yaml b/training/configs/ornith-35b-lora.yaml similarity index 100% rename from training/configs/llama2-7b-lora.yaml rename to training/configs/ornith-35b-lora.yaml