Skip to content

Commit 4ca7cfc

Browse files
committed
fix lora_finetune variant logic
Signed-off-by: nvmvle <mvle@nvidia.com>
1 parent c1f71ff commit 4ca7cfc

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

ci/benchmarks/partial-conv/evo2_finetuning.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,19 +44,19 @@ script_args:
4444
precision: fp8
4545
products:
4646
- variant: finetune
47-
lora_enabled: false
47+
lora_enabled: ""
4848
task: finetune_from_ckpt
4949
experiment_name: evo2-finetune
5050
- variant: lora_finetune
51-
lora_enabled: true
51+
lora_enabled: "--lora-finetune"
5252
task: lora_finetune_from_ckpt
5353
experiment_name: evo2-lora-finetune
5454
script: |-
5555
WANDB_API_KEY=$BIONEMO_WANDB_API_KEY train_${model} \
5656
-d ${data_base_path}/${dataset_config} \
5757
--dataset-dir=${data_base_path}/${dataset_dir} \
5858
--ckpt-dir=${data_base_path}/${restore_from_checkpoint_path} \
59-
${ if lora_enabled then "--lora-finetune" else "" } \
59+
${lora_enabled} \
6060
--model-size=${model_size} \
6161
--max-steps=${max_steps} \
6262
--experiment-name=${experiment_name}_${batch_size}bs_${nodes}node_${gpus}gpu_${max_steps}s \

0 commit comments

Comments
 (0)