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

Kosmos-2 batch modality and processing speed #1604

Open
unrue opened this issue Jul 18, 2024 · 0 comments
Open

Kosmos-2 batch modality and processing speed #1604

unrue opened this issue Jul 18, 2024 · 0 comments

Comments

@unrue
Copy link

unrue commented Jul 18, 2024

I'm using Kosmos-2 referring to

https://huggingface.co/microsoft/kosmos-2-patch14-224

using a simple prompt "An image of".

The process works well but it is quite slow. Over 1000 images, it requires 3 hours and half on Nvidia A100 GPU. Is it possible to enable batch modality? At the moment, one image per time is processed.

Other hints to improve processing speed? Thanks.

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

No branches or pull requests

1 participant