torch2trt icon indicating copy to clipboard operation
torch2trt copied to clipboard

Support text models

Open aliencaocao opened this issue 1 year ago • 0 comments

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 https://github.com/NVIDIA-AI-IOT/torch2trt/issues/931

Closes https://github.com/NVIDIA-AI-IOT/torch2trt/issues/931

Tested with SigLIP and results matches closely,

aliencaocao avatar Jun 08 '24 07:06 aliencaocao