czczup's picture kosung's picture
Fix InternLM2ForCausalLM does not support Flash Attention 2.0 yet (#3)
743a544 verified