diff --git a/recipes/configs/llama3_2/3B_qlora_single_device.yaml b/recipes/configs/llama3_2/3B_qlora_single_device.yaml index a98e1ac9c..c9c4a57ff 100644 --- a/recipes/configs/llama3_2/3B_qlora_single_device.yaml +++ b/recipes/configs/llama3_2/3B_qlora_single_device.yaml @@ -69,7 +69,7 @@ loss: epochs: 1 max_steps_per_epoch: null gradient_accumulation_steps: 4 -compile=False # pytorch compile, set to true for perf/memory improvement# set it to True for better memory and performance +compile: False # pytorch compile, set to true for perf/memory improvement# set it to True for better memory and performance # Logging output_dir: /tmp/lora_finetune_output