search

unitaryai / detoxify

Trained models & code to predict toxic comments on all 3 Jigsaw Toxic Comment Challenges. Built using ⚡ Pytorch Lightning and 🤗 Transformers. For access to our API, please email us at contact@unitary.ai.
https://www.unitary.ai/
Apache License 2.0
968 stars 114 forks source link

fix for transformers 4.35.2 #98

Closed Vela-zz closed 10 months ago

Vela-zz commented 12 months ago

latest transformers do not initial model_class with arguments directly anymore,