Fast and easy toolkit for distilling AI models
Project description
QuickDistill
Fast and easy toolkit for evaluating AI models (and training them in the future)
QuickDistill provides an intuitive web UI for the complete model distillation workflow:
- 📊 View and filter Weave traces from your LLM calls
- 🎯 Export strong model outputs as test sets
- 🔬 Run weak models on strong outputs
- ⚖️ Evaluate similarity using LLM judges
- 📥 Download evaluation datasets for analysis
Installation
pip install quickdistill
Quick Start
Launch the UI in your current directory:
quickdistill launch
This will:
- Start the QuickDistill server on
http://localhost:5001 - Open the Trace Viewer in your browser
- Create local directories for projects, exports, and results
Requirements
Set these environment variables:
export WANDB_API_KEY="your_wandb_key" # Required for W&B Inference
export OPENROUTER_API_KEY="your_openrouter_key" # Optional for OpenRouter models
Get your keys:
- W&B: https://wandb.ai/settings
- OpenRouter: https://openrouter.ai/keys
Usage
1. Fetch Weave Traces
Enter your Weave project name (e.g., username/project-name) and click "Fetch New Project" to load traces from W&B.
2. Export Strong Model Outputs
- Filter traces by model or operation
- Select traces to use as ground truth
- Export to create a test set
3. Run Weak Models
- Select a strong model export
- Choose W&B models from the list or enter custom OpenRouter models
- Run inference to generate weak model outputs
4. Create Judges
Navigate to /judge to create LLM judges:
- Scalar judges: Rate similarity on a numeric scale (1-5)
- Boolean judges: Determine if outputs are correct/incorrect
5. Run Evaluations
- Select weak model results
- Choose a judge
- Run evaluation and view results in Weave
6. Download Datasets
Click "Download" next to any weak model result to get a clean JSON dataset with:
[
{
"input": "question text...",
"strong_model": "model-name",
"strong_output": "strong response...",
"weak_model": "model-name",
"weak_output": "weak response..."
}
]
CLI Options
quickdistill launch # Launch on default port 5001
quickdistill launch --port 8080 # Launch on custom port
quickdistill launch --no-browser # Don't open browser automatically
quickdistill launch --debug # Run in debug mode
Project Structure
When you run quickdistill launch, it creates these directories ~/.cache/quickdistill:
your-project/
├── projects/ # Cached Weave traces by project
│ └── username_project/
│ └── traces_data.json
├── strong_exports/ # Exported strong model test sets
│ └── model-name_10traces.json
├── weak_model_*.json # Weak model inference results
├── judges.json # Saved judge configurations
└── evaluations/ # Evaluation results
Features
- Multi-provider support: Works with W&B Inference and OpenRouter
- Flexible judging: Create custom LLM judges or use pre-built ones
- Trace filtering: Filter by model, operation, or custom criteria
- Batch operations: Run multiple models and evaluations in parallel
- Export formats: Download clean datasets for further analysis
- Project isolation: Each Weave project is cached separately
Development
Install in development mode:
git clone https://github.com/byyoung3/quickdistill.git
cd quickdistill
pip install -e .
License
MIT
Author
Brett Young (bdytx5@umsystem.edu)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file quickdistill-0.1.13.tar.gz.
File metadata
- Download URL: quickdistill-0.1.13.tar.gz
- Upload date:
- Size: 3.4 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
c022220c79a78a728fe05a3321559fe8f98889847477e2b31cc9cd98498d7414
|
|
| MD5 |
c0ed4edae25dadf072971eab75fedc6e
|
|
| BLAKE2b-256 |
3b5a0bdf853dbf7f17ca7c4b4ce53ffd6dc2e386cabf54038d9f93bb4768b75b
|
File details
Details for the file quickdistill-0.1.13-py3-none-any.whl.
File metadata
- Download URL: quickdistill-0.1.13-py3-none-any.whl
- Upload date:
- Size: 904.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
233bff74ec539997fed55ce9a9179ebc97d30575bb7e4a8faaae75b9de8bdc2c
|
|
| MD5 |
f9fa672a86e55524206bf6addbd5e71f
|
|
| BLAKE2b-256 |
5a1fabefd4e18bfbaeba903daeb96abc6a6a4b67655241fd8893826a0024461d
|