Skip to content

Commit 4fd16cd

Browse files
huvunvidiaAmir Hussein
authored andcommitted
Update AVLM finetune example for vanilla fine-tuning (NVIDIA-NeMo#14232)
Signed-off-by: Huy Vu <86480512+huvunvidia@users.noreply.github.com> Signed-off-by: Amir Hussein <amhussein@nvidia.com>
1 parent d7fd100 commit 4fd16cd

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

scripts/avlm/avlm_nemo_run.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,7 @@ def configure_recipe(
4646
num_nodes=nodes,
4747
num_gpus_per_node=gpus_per_node,
4848
freeze_modules=freeze_modules,
49-
# DEBUGGING
50-
peft_scheme="lora",
49+
peft_scheme="none",
5150
)
5251
recipe.trainer.max_steps = 20
5352
recipe.trainer.val_check_interval = 20

0 commit comments

Comments
 (0)