update wan configs to use default 512 flash block sizes.

Author: entrpn

src/maxdiffusion/configs/base_wan_14b.yml

@@ -73,15 +73,7 @@ mask_padding_tokens: True
 attention_sharding_uniform: True 
 dropout: 0.1
 
-flash_block_sizes: {
-  "block_q" : 2048,
-  "block_kv_compute" : 512,
-  "block_kv" : 2048,
-  "block_q_dkv" : 2048,
-  "block_kv_dkv" : 2048,
-  "block_kv_dkv_compute" : 512,
-  "use_fused_bwd_kernel": True
-}
+flash_block_sizes: {}
 # Use on v6e
 # flash_block_sizes: {
 #   "block_q" : 3024,

src/maxdiffusion/configs/base_wan_27b.yml

@@ -72,16 +72,7 @@ mask_padding_tokens: True
 attention_sharding_uniform: True 
 dropout: 0.1
 
-flash_block_sizes: {
-  "block_q" : 1024,
-  "block_kv_compute" : 256,
-  "block_kv" : 1024,
-  "block_q_dkv" : 1024,
-  "block_kv_dkv" : 1024,
-  "block_kv_dkv_compute" : 256,
-  "block_q_dq" : 1024,
-  "block_kv_dq" : 1024
-}
+flash_block_sizes: {}
 # Use on v6e
 # flash_block_sizes: {
 #   "block_q" : 3024,