Irene Dea

Results 2 comments of Irene Dea

@vtien @getao Do you get the CheckpointError with `use_flash_attention_2=False`?

@XiaohanZhangCMU do you need a review on this PR?