Vulntrain

Latest version: v1.2.0

Safety actively analyzes 723650 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

Page 2 of 2

0.3.0

News

Dataset generation: allow specifying a commit message when uploading to Hugging Face.

Validation: Added a simple validation script for model optimized for text generation. The script is
able to pull a model and send tasks via a Pipeline

Changes

For the training step: added the choices of model: gpt2, distilgpt2,
meta-llama/Llama-3.3-70B-Instruct, distilbert-base-uncased

Various improvements to the command line parsing.

0.2.0

News

Added a trainer.
Experimenting distilbert-base-uncased (AutoModelForMaskedLM) and gpt2 (AutoModelForCausalLM).
The goal is to generate text.

Changes

Various improvements to the dataset generator. And added a command line parser.

0.1.0

First release with upload of datasets to HuggingFace.

Datasets are build based on NIST data with enrichment from FKIE and vulnrichment

Page 2 of 2

© 2025 Safety CLI Cybersecurity Inc. All Rights Reserved.