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

[REQUEST] Deepspeed Inference Supports VL (vision language) model #6917

Open
ethen8181 opened this issue Dec 26, 2024 · 0 comments
Open

[REQUEST] Deepspeed Inference Supports VL (vision language) model #6917

ethen8181 opened this issue Dec 26, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@ethen8181
Copy link

Is your feature request related to a problem? Please describe.
We have been using deepspeed.init_inference API for speeding up inference for text only models (e.g. mistral, qwen 2.5 series) with success. Was hoping we can extend support for vision language models as well, e.g. qwen 2 vl, etc, which is currently not supported.

Describe the solution you'd like

  • deepspeed.init_inference to work for vision language models (for both embedding use case as well as generation use case)
  • and also make extending with our own model's tutorial clearer/cleaner.

Describe alternatives you've considered
N/A

Additional context
N/A

@ethen8181 ethen8181 added the enhancement New feature or request label Dec 26, 2024
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

1 participant