Error running on exported onnx model

[yaphet266]

The export of ckpt file to onnx model resulted in an error message, as shown below：
[ErrorCode:NotImplemented] Could not find an implementation for Trilu(14) node with name '/Trilu'"

System information：
onnxruntime:1.13.1
opset_version=14
torch 2.2.1+cu118

[hungcao0402]

I changed dtype=torch.bool to torch.float and success

parseq/strhub/models/parseq/model.py

Line 117 in 1902db0

tgt_mask = query_mask = torch.triu(torch.ones((num_steps, num_steps), dtype=torch.bool, device=self._device), 1)