search

QData / TextAttack

TextAttack 🐙 is a Python framework for adversarial attacks, data augmentation, and model training in NLP https://textattack.readthedocs.io/en/master/
https://textattack.readthedocs.io/en/master/
MIT License
2.98k stars 397 forks source link

Update model_args.py #706

Closed plasmashen closed 2 years ago

plasmashen commented 2 years ago

the 'model' is a T5ForTextToText class instead of transformers.PreTrainedModel class which can not be processed by HuggingFaceModelWrapper.