---
license: cc-by-nc-sa-4.0
datasets:
- taigasan/e6-visual-ratings
base_model:
- lodestones/tagger-experiment
---
# Aesthetic Classifier Trainer

Community:

https://discord.gg/nm3mhfhP


Quick start:

1. source venv/bin/activate
2. pip install -r requirements.txt
3. WANDB_API_KEY=<your_wandb_api_key> python train.py

Notes:

- tagger_proto.safetensors came from https://huggingface.co/lodestones/tagger-experiment
- embed_cache.pt is included so training can run turn-key without fetching local samples for embedding cache build.

Credits:
- All community members who are contributing ratings via https://taigasan-e6-visual-ratings.hf.space/
- Thessalo https://huggingface.co/Thouph
- Taigasan https://huggingface.co/taigasan