File size: 686 Bytes
b7720f0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
install:
	python -m pip install -r requirements.txt

train:
	python train.py --qlora --bf16 --num_train_epochs 3 --train_batch_size 1 --gradient_accumulation_steps 16

infer:
	python infer.py --model_dir outputs/GravityLLM-Qwen2.5-1.5B-S9 --input_json examples/sample_input.json --validate --output_json outputs/sample_prediction.json

eval:
	python evaluate.py --model_dir outputs/GravityLLM-Qwen2.5-1.5B-S9 --data_file data/valid.jsonl --report_path reports/eval_report.json

validate-example:
	python tools/validate_scene.py schemas/scene.schema.json examples/sample_output.json

synthetic-data:
	python tools/make_synthetic_dataset.py --output data/synthetic_train.jsonl --count 50