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

Torchscript Model can't have bfloat16 inputs / outputs in 24.09 #7853

Open
MatthieuToulemont opened this issue Dec 5, 2024 · 0 comments
Open

Torchscript Model can't have bfloat16 inputs / outputs in 24.09 #7853

MatthieuToulemont opened this issue Dec 5, 2024 · 0 comments

Comments

@MatthieuToulemont
Copy link

I am using the triton container from 24.09 and want to serve torscript models with BFLOAT16 inputs but I am getting the following error:

failed to load 'UnflattenSequence' version 1: Internal: unsupported datatype TYPE_BF16 for input 'latent_as_sequence_input__0' for model 'UnflattenSequence'"

Given that torchscript supports bfloat16, shouldn't this be supported as well ?
@MatthieuToulemont MatthieuToulemont changed the title Torchscript Model can't have bfloat16 inputs in 24.09 Torchscript Model can't have bfloat16 inputs / outputs in 24.09 Dec 5, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@MatthieuToulemont