[NEW MODEL] - SupraSafety-18M · Tiny Content-Moderation Model

Hey r/LocalLLaMA !

SupraLabs is back with a new model: SupraSafety-18M.

It's a BERT-style 18M params model trained from scratch on 2 T4 GPUs in Kaggle on the nvidia/Nemotron-3.5-Content-Safety-Dataset dataset for 7 epochs.

It's built to run on edge devices, mobile phones, or in low-latency production environments for content moderation.

this model is a binary text classifier with the classes SAFE and UNSAFE.

Here are a few examples on how it performs:

The model reached an acuracy of 81.2% and a precision of 86.9% after the full training run.

Link to the model: https://huggingface.co/SupraLabs/SupraSafety-18M

SupraLabs on Hugging Face (give us a follow if you like what we are doing ❤️🤗): https://huggingface.co/SupraLabs

Feel free to use it, test it, give honest feedback, etc. We read every comment!

Thanks for your interest in our models!

CU with the next model! 🤩