search

LAION-AI / aesthetic-predictor

A linear estimator on top of clip to predict the aesthetic quality of pictures
MIT License
420 stars 20 forks source link

ViT-B/16 pre-computed embeddings? #2

Open apolinario opened 2 years ago

apolinario commented 2 years ago

Since you have the ViT-B/16 aesthetic model, was just wondering whether you would have the ViT-B/16 pre-computed embeddings too

rom1504 commented 2 years ago

I don't have ViT-B/16 embeddings over laion2b or laion400m so I would need to compute some (10M is enough to do that aggregation)

Is B/16 useful?

apolinario commented 2 years ago

It would be. I have already hooked the ViT-B/32 and 14 embeddings to Majesty Diffusion, and ViT-B/16 is very useful on image generation as it's still one of the most used for quality vs. vram usage purposes

rom1504 commented 2 years ago

Ok will get that