[hotfix] fix implement error in diffusers

[1SAA]

What's New

Fix NotImplementedError: Some torch function is incompatible because of its complcated inputs. when training diffusers.

• add a ignore step for no grad tensors
• change the original checkpoint function in diffuser to torch checkpint function