Update 3B_qlora_single_device.yaml

pytorch · Oct 25, 2024 · c0e9645 · c0e9645
1 parent 41b0012
commit c0e9645
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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