Distilabel is a framework for synthetic data and AI feedback for engineers who need fast, reliable and scalable pipelines based on verified research papers.
Distilabel is a framework for synthetic data and AI feedback for engineers who need fast, reliable and scalable pipelines based on verified research papers.
[!IMPORTANT]
The original authors have moved on to other projects. A group of community members have recently joined the GitHub project as collaborators to maintain the project and are actively working towards the next release. Check out the develop branch for access to the latest fixes and improvements in the meantime.
Synthesize data for AI and add feedback on the fly!
Distilabel is the framework for synthetic data and AI feedback for engineers who need fast, reliable and scalable pipelines based on verified research papers.
If you just want to get started, we recommend you check the documentation. Curious, and want to know more? Keep reading!
<!-- -->
Why use distilabel?
Distilabel can be used for generating synthetic data and AI feedback for a wide variety of projects including traditional predictive NLP (classification, extraction, etc.), or generative and large language model scenarios (instruction following, dialogue generation, judging etc.). Distilabel's programmatic approach allows you to build scalable pipelines for data generation and AI feedback. The goal of distilabel is to accelerate your AI development by quickly generating high-quality, diverse datasets based on verified research methodologies for generating and judging with AI feedback.
Improve your AI output quality through data quality
Compute is expensive and output quality is important. We help you focus on data quality, which tackles the root cause of both of these problems at once. Distilabel helps you to synthesize and judge data to let you spend your valuable time achieving and keeping high-quality standards for your data.
Take control of your data and models
Ownership of data for fine-tuning your own LLMs is not easy but Distilabel can help you to get started. We integrate AI feedback from any LLM provider out there using one unified API.
Improve efficiency by quickly iterating on the right research and LLMs
Synthesize and judge data with latest research papers while ensuring flexibility, scalability and fault tolerance. So you can focus on improving your data and training your models.
Community
We are an open-source community-driven project and we love to hear from you. Here are some ways to get involved:
Community Meetup: listen in or present during one of our bi-weekly events.
Discord: get direct support from the community in #argilla-general and #argilla-help.
Roadmap: plans change but we love to discuss those with our community so feel encouraged to participate.
What do people build with Distilabel?
The Argilla community uses distilabel to create amazing datasets and models.
The 1M OpenHermesPreference is a dataset of ~1 million AI preferences derived from teknium/OpenHermes-2.5. It shows how we can use Distilabel to synthesize data on an immense scale.
hf-transformers: for using models available in transformers package via the TransformersLLM integration.
litellm: for using LiteLLM to call any LLM using OpenAI format via the LiteLLM integration.
llama-cpp: for using llama-cpp-python Python bindings for llama.cpp via the LlamaCppLLM integration.
mistralai: for using models available in Mistral AI API via the MistralAILLM integration.
ollama: for using Ollama and their available models via OllamaLLM integration.
openai: for using OpenAI API models via the OpenAILLM integration, or the rest of the integrations based on OpenAI and relying on its client as AnyscaleLLM, AzureOpenAILLM, and TogetherLLM.
vertexai: for using Google Vertex AI proprietary models via the VertexAILLM integration.
vllm: for using vllm serving engine via the vLLM integration.
sentence-transformers: for generating sentence embeddings using sentence-transformers.
mlx: for using MLX models via the MlxLLM integration.
Structured generation
outlines: for using structured generation of LLMs with outlines.
instructor: for using structured generation of LLMs with Instructor.
Data processing
ray: for scaling and distributing a pipeline with Ray.
faiss-cpu and faiss-gpu: for generating sentence embeddings using faiss.
text-clustering: for using text clustering with UMAP and Scikit-learn.
minhash: for using minhash for duplicate detection with datasketch and nltk.
Example
To run the following example you must install distilabel with the hf-inference-endpoints extra:
sh
pip install "distilabel[hf-inference-endpoints]" --upgrade
Then run:
```python
from datasets import load_dataset
from distilabel.models import InferenceEndpointsLLM
from distilabel.pipeline import Pipeline
from distilabel.steps.tasks import TextGeneration
with Pipeline() as pipeline:
TextGeneration(
llm=InferenceEndpointsLLM(
modelid="meta-llama/Meta-Llama-3.1-8B-Instruct",
generationkwargs={"temperature": 0.7, "maxnewtokens": 512},
),
)
if name == "main":
dataset = loaddataset("distilabel-internal-testing/instructions", split="test")
distiset = pipeline.run(dataset=dataset)
distiset.pushtohub(repoid="distilabel-example")
```
Badges
If you build something cool with distilabel consider adding one of these badges to your dataset or model card.
[<img src="https://raw.githubusercontent.com/argilla-io/distilabel/main/docs/assets/distilabel-badge-light.png" alt="Built with Distilabel" width="200" height="32"/>](https://github.com/argilla-io/distilabel)
[<img src="https://raw.githubusercontent.com/argilla-io/distilabel/main/docs/assets/distilabel-badge-dark.png" alt="Built with Distilabel" width="200" height="32"/>](https://github.com/argilla-io/distilabel)
bibtex
@misc{distilabel-argilla-2024,
author = {Álvaro Bartolomé Del Canto and Gabriel Martín Blázquez and Agustín Piqueres Lajarín and Daniel Vila Suero},
title = {Distilabel: An AI Feedback (AIF) framework for building datasets with and for LLMs},
year = {2024},
publisher = {GitHub},
journal = {GitHub repository},
howpublished = {\url{https://github.com/argilla-io/distilabel}}
}
cff-version: 1.2.0
message: "If you use this software, please cite it as below."
authors:
- family-names: "Bartolomé"
given-names: "Álvaro"
- family-names: "Martín-Blázquez"
given-names: "Gabriel"
- family-names: "Piqueres-Lajarín"
given-names: "Agustín"
- family-names: "Vila-Suero"
given-names: "Daniel"
title: "Distilabel: An AI Feedback (AIF) framework for building datasets with and for LLMs."
version: 1.1.1
date-released: 2024-05-22
url: "https://github.com/argilla-io/distilabel"
