pharmapsychotic / clip-interrogator

Image to prompt with BLIP and CLIP
MIT License
2.71k stars 431 forks source link

SDXL Clip Support #86

Open adammenges opened 1 year ago

adammenges commented 1 year ago

Going to go ahead and make an issue for SDXL, will be an obvious request here in the next couple weeks.

https://github.com/pharmapsychotic/clip-interrogator/blob/2cf03aaf6e704197fd0dae7c7f96aa59cf1b11c9/predict.py#L20-L24

rossaai commented 1 year ago

SDXL uses OpenClip "ViT-g-14/laion2B-s34B-b88K"

config = Config()
config.clip_model_name = "ViT-g-14/laion2B-s34B-b88K"
model = Interrogator(config)