pharmapsychotic / clip-interrogator

Image to prompt with BLIP and CLIP

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Image batch processing?

kaijingxjtu opened this issue · comments

Thanks for the great work!

Could this program run faster on image batches using a Dataloader Class or sth ?

And plus, could this project run on muti-GPUs?

You can modify the code to extract image/text features in batches but that won't yield any significant speed boost. Bottleneck is the flavor chaining part. I'm not sure that can be done in batches or not.