docs: rename config and docs to reflect Ornith-35B model
This commit is contained in:
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user