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

[Feature] pipeline.get_logits() support tensor on gpu as input rather than list input #2926

Open

BradZhone opened this issue Dec 19, 2024 · 0 comments

Assignees

BradZhone commented Dec 19, 2024

Motivation

Now the get_logits() function of pipeline only support list input from cpu, could you support tensor input on gpu?

Related resources

No response

Additional context

No response

lvhan028 assigned lzhangzz

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment