Aesthetic Classifier Trainer
Community:
Quick start:
- source venv/bin/activate
- pip install -r requirements.txt
- 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
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
Model tree for RedRocket/e6-visual-ratings
Base model
lodestones/tagger-experiment