Using gradient_checkpointing_kwargs={'use_reentrant':False} instead of gradient_checkpointing=False might work.
gradient_checkpointing_kwargs={'use_reentrant':False}
gradient_checkpointing=False