WeightDrop Attribute Error for NLP text generation

Hey folks,

Has anyone run into this attribute error when testing the model on flask?