All repositories

Securityprotectai

modelscan

A scanner for model files that detects serialization attacks, such as malicious code inside a pickle. It checks the file before you load it.

Who it is for

For people who download third-party models from hubs and want to avoid running malicious code.

How to start

  1. Install: pip install modelscan
  2. Scan a model file: modelscan -p /path/to/model_file.pkl
  3. Read the report and do not load the model if issues are found

Steps are taken from the README. Check the current version in the repository before running them.

Stars over the last 30 days

+10Sep 6 — Sep 28
768778

Author's description

Protection against Model Serialization Attacks

data-privacy-stack

presidio

A framework that finds and masks personal data in text, images and tables. It detects the data, then redacts or replaces it so it can be safely sent to a model.

11Kstars+430 in 30 dPython

NVIDIA

garak

A vulnerability scanner for LLMs: it runs a model through a set of probes for prompt injection, data leakage, toxicity and other failures. Works as a command-line tool.

9.4Kstars+325 in 30 dPython

guardrails-ai

guardrails

A Python framework that checks LLM inputs and outputs with ready validators from Guardrails Hub and helps produce structured answers. It catches risks before a reply reaches the user.

7.5Kstars+137 in 30 dPython

NVIDIA-NeMo

Guardrails

NVIDIA's toolkit for programmable rails in LLM-based conversational systems. Rules live in config and control what the bot talks about and does.

7.2Kstars+183 in 30 dPython

microsoft

PyRIT

Microsoft's framework for proactively finding risks in generative AI systems. It helps automate red teaming against models and applications.

4.6Kstars+174 in 30 dPython

meta-llama

PurpleLlama

Meta's generative AI safety project: the Llama Guard and Prompt Guard filter models, the Code Shield scanner and the CyberSec Eval test suites. It pairs defense with attack-based testing.

4.4Kstars+45 in 30 dPython