bugfix: Update modeling_t5.T5Stack.forward() for Gradient Checkpointing

by Panda-vid - opened Jan 22

←

Jan 22

Update checkpoint() call such that parameters for the layer_module object are passed correctly.

plenz changed pull request status to closed Jan 25

plenz

Owner Jan 25

The feature only works with older transformer versions

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment