Text Generation
Transformers
PyTorch
Safetensors
English
stripedhyena
custom_code

checkpoint() method is missing

#4
by kashif - opened

when gradient_checkpointing is true there is no checkpoint method defined in the stateless_forward

Zymrael changed discussion status to closed

Sign up or log in to comment