Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Support text models #932

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Conversation

aliencaocao
Copy link

@aliencaocao aliencaocao commented Jun 8, 2024

Because how flattener limits to floats as inputs, it will wrongly remove any int inputs that are common in text models e.g. encoded token ids.

Discovered when trying to convert a text transformer for SigLIP in #931

Closes #931

Tested with SigLIP and results matches closely,

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

Successfully merging this pull request may close these issues.

Unable to convert SigLIP text transformer due to missing model input when exporting model to onnx
1 participant