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