Lora misconfiguration

This commit is contained in:
Jake Poznanski 2024-09-25 10:48:39 -07:00
parent 86813fe210
commit 3a5b438a6f
2 changed files with 2 additions and 2 deletions

View File

@ -84,6 +84,6 @@ lora:
save:
path: s3://ai2-oe-data/jakep/experiments/qwen2vl-pdf/v1/models/
save_every_steps: 500
save_every_steps: 1000
max_workers: 10

View File

@ -10,7 +10,7 @@ then
fi
EXTRA_ARGS="-c pdelfin/train/config/qwen2vl-2b.yaml --num_proc 64 --save.path \"s3://ai2-oe-data/jakep/experiments/qwen2vl-pdf/v1/models/\${BEAKER_USER_ID}\""
EXTRA_ARGS="-c pdelfin/train/config/qwen2vl-2b-lora.yaml --num_proc 64 --save.path \"s3://ai2-oe-data/jakep/experiments/qwen2vl-pdf/v1/models/\${BEAKER_USER_ID}\""
run_name=$(basename "$0" .sh)