Wrong commands in compiler_toolkit .md?

[yushangdi]

Bug description

The commands in the readme page of https://github.com/pytorch/torchtitan/tree/main/torchtitan/experiments/compiler_toolkit are wrong?
Only the first flex_attention command has --model.flavor=debugmodel_flex_attn, the other three don't, and I don't see flex_attention ops in the graph modules if I don't specify the model.flavor.

Versions

main 1bd2548

[yushangdi]

cc @yiming0416 do you know if the commands are wrong? just want to make sure I'm running the right commands

[yiming0416]

@yushangdi you should always add --model.flavor=debugmodel_flex_attn in order to have flex attention in the model.

I think I forgot to add it in some of the commands in REAMDE.