Error Transfer to JIT or ONNX

[ohhyeeaah]

Hi there, really great job! I want to save the pretrained COVER model to a lower resource inference environment for inference, and I meet with frustrating error message like this:

it occurs when excuting CLIPIQA.forward:

My model is init as :

Thanks !