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
893 stars 115 forks source link

Allow newer torch for numpy 2.0 compatibility #109

Open jamt9000 opened 4 weeks ago

jamt9000 commented 4 weeks ago

Remove the torch upper bound, to allow installing a version compatible with numpy 2.0