WIP dont use flash attention

ecmwf · Feb 21, 2025 · c0b206b · c0b206b
1 parent 924b1fa
commit c0b206b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/basic_config.yaml b/tests/basic_config.yaml
@@ -8,7 +8,7 @@ defaults:
 - training: default
 - _self_
 
-
+no_validation: True
 # diagnostics:
 #   plot:
 #     callbacks: []
@@ -30,6 +30,8 @@ hardware:
 
 model:
   num_channels: 16
+  processor:
+    attention_implementation: scaled_dot_product_attention
 
 dataloader:
   limit_batches: