Transformers
PyTorch
code
custom_code
Inference Endpoints

Included gradient checkpointing

#1
by FJFehr - opened

This is a minor change that allows for gradient checkpointing. This allows for increased batch sizes when file-tuning these models.

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment