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.

Sign up for GitHub

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

Jump to bottom

Do not use pipelines sequentially on GPU for zero-shot calculation #176

Open
FelixKirsch opened this issue Nov 15, 2022 · 0 comments
Open

Do not use pipelines sequentially on GPU for zero-shot calculation #176

FelixKirsch opened this issue Nov 15, 2022 · 0 comments
Labels
enhancement New feature or request

Comments

@FelixKirsch
Copy link
Contributor

Is your feature request related to a problem? Please describe.
When running a zero-shot model on GPU a warning is emitted.
Screenshot 2022-11-15 at 18 01 53
The zero-shot services run one pipeline after another on batches of records. This is not optimal and decreases the performance of the calculation.

Describe the solution you'd like
The huggingface documentation describes different ways to implement a solution so that the pipelines are run in parallel.
@FelixKirsch FelixKirsch added the enhancement New feature or request label Nov 15, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@FelixKirsch