Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature request] Additional scorer options for auto merger #126

Open
Enferlain opened this issue Mar 8, 2024 · 2 comments
Open

[Feature request] Additional scorer options for auto merger #126

Enferlain opened this issue Mar 8, 2024 · 2 comments
Assignees
Labels
enhancement New feature or request

Comments

@Enferlain
Copy link

There are a few potential scorers that could be worth a try

https://huggingface.co/shadowlilac/aesthetic-shadow-v2 - Anime. Large model, maybe needs to be converted to fp16. Was used for the Animagine v3 datasets
https://huggingface.co/Eugeoter/waifu-scorer-v2 - Anime. Small. By one of the better anime finetune makers. Haven't tested yet.
https://huggingface.co/yuvalkirstain/PickScore_v1 - Large model, not 100% sure about the function but I've used it before.

Idea from here along with some other implementations like BLIP CLIP hpsv2 etc.

Maybe could run multiple scorers together and average them out to have more opinions. And it could also be nice to implement a manual scoring option, although that wouldn't strictly fall under "auto merging" I guess, but it would be useful nonetheless.

@Enferlain Enferlain changed the title [Feature request] Additional scorers options for auto merger [Feature request] Additional scorer options for auto merger Mar 8, 2024
@wkpark wkpark self-assigned this Mar 9, 2024
wkpark added a commit that referenced this issue Mar 12, 2024
@wkpark wkpark added the enhancement New feature or request label Mar 12, 2024
@wkpark
Copy link
Owner

wkpark commented Mar 13, 2024

pick scorer_v1 + aesthetic_shadow_v2 scorers added.

  • aesthetic shadow ~ 4.3GB
  • pick scorer v1 ~ 3.7GB

@Koitenshin
Copy link

@wkpark You added the original shadow scorer which gives problematic scores, hence the v2.

I tested multiple images, both viable and messed up ones, including some that were 80% noise. Every single image was given a 0.99+ score.

@Enferlain I was looking for scorers myself so I just made a guide on how to do it.

##148

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

3 participants