Downloads · 30 days
374
1% of all-time downloads
HPSv2-hf
adams-story/HPSv2-hf
HPSv2-hf is a zero-shot image classification model from adams-story. Use it for the zero-shot image classification task on the model card, and read the license before you ship it in a product. It is set up for transformers.
This is a Huggingface CLIPModel flavor of the HPSv2 model, which is trained to predict human preferences over AI generated images.
All-time downloads
35.1K
Public
Repo size
3.9 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin2 GB · 100%
At a glance
- Task
- Zero-Shot Image Classification
- Library
- transformers
- Model type
- clip
- Access
- Public
- Created
- Sep 19, 2023
- Updated
- Nov 19, 2023
- SHA
- f535ee1d
Model card
From the Hugging Face model README
HPSv2-hf
This is a Huggingface CLIPModel flavor of the HPSv2 model, which is trained to predict human preferences over AI generated images.
I converted the model weights from the openclip format to huggingface CLIPModel. The two text and image embeddings were tested to be equal before and after conversion.
You can load the model the same as any huggingface clip model:
from transformers import CLIPProcessor, CLIPModel
model = CLIPModel.from_pretrained("adams-story/HPSv2-hf")
processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32") # uses the same exact vanilla clip processor
All credit goes to the original authors of HPSv2
- Task
- Zero-Shot Image Classification
- Library
- transformers
- Type
- clip
- Created
- Sep 19, 2023
- Updated
- Nov 19, 2023