fix qwen2 merge

ebsmothers · ebsmothers · commit ba1cb9283e8c · 2024-08-10T11:28:32.000-07:00
diff --git a/recipes/configs/qwen2/0.5B_full.yaml b/recipes/configs/qwen2/0.5B_full.yaml
@@ -60,7 +60,7 @@ gradient_accumulation_steps: 16
 device: cuda
 
 # Memory management
-enable_activation_checkpointing: True
+enable_activation_checkpointing: False
 
 # Reduced precision
 dtype: bf16
diff --git a/recipes/configs/qwen2/1.5B_full.yaml b/recipes/configs/qwen2/1.5B_full.yaml
@@ -60,7 +60,7 @@ gradient_accumulation_steps: 1
 device: cuda
 
 # Memory management
-enable_activation_checkpointing: True
+enable_activation_checkpointing: False
 
 # Reduced precision
 dtype: bf16
