Duplicated from hugging-apps/metaview
ccfee12
1
2
from .gradient_checkpoint import gradient_checkpoint_forward
