Downloads · 30 days
62
7% of all-time downloads
pixai-tagger-v1.0-mixed-bf16
DraconicDragon/pixai-tagger-v1.0-mixed-bf16
pixai-tagger-v1.0-mixed-bf16 is a image classification model from DraconicDragon. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
BF16 mixed precision conversion of https://huggingface.co/pixai-labs/pixai-tagger-v1.0. Refer to original repo for all the info. head. tensors are kept in FP32.
All-time downloads
953
Public
Parameters
486M
1 GB on disk
Likes
4
Public
Repo makeup
Click a slice to open those files.
.safetensors1 GB · 100%
Parameter types
How the weights are stored.
BF16455M · 93%
Base models
Model card
From the Hugging Face model README
BF16 mixed precision conversion of https://huggingface.co/pixai-labs/pixai-tagger-v1.0. Refer to original repo for all the info.
head.* tensors are kept in FP32.
tagger_pipeline.py in this repo also contains a 2 line change in the last def forward(self, x): to cast x to head's weight dtype which should prevent mismatch if autocast is disabled
- Task
- Image Classification
- Library
- transformers
- Type
- cls_vitdet
- License
- apache-2.0
- Created
- Sep 19, 2026
- Updated
- Sep 23, 2026