layer head mask error
shivanisrivarshini opened this issue · comments
Error is from custom_t5.py,line 256
TypeError: forward() got an unexpected keyword argument 'layer_head_mask'
Fraser-Greenlee / transformer-vae
A library for making Transformer Variational Autoencoders. (Extends the Huggingface/transformers library.)